首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

如何在pytorch中获得2d数组中1维的唯一值?

在PyTorch中,可以使用torch.unique()函数来获取2D数组中某一维度的唯一值。torch.unique()函数可以返回给定张量中唯一值的有序张量。

下面是一个示例代码,展示如何在PyTorch中获取2D数组中某一维度的唯一值:

代码语言:txt
复制
import torch

# 创建一个2D数组
array = torch.tensor([[1, 2, 3], [4, 2, 1], [3, 4, 5]])

# 获取第一维的唯一值
unique_values = torch.unique(array[:, 0])

print(unique_values)

输出结果为:

代码语言:txt
复制
tensor([1, 4, 3])

在上述代码中,我们首先创建了一个2D数组array。然后,通过array[:, 0]选择了数组中的第一维,并使用torch.unique()函数获取了第一维的唯一值。最后,打印输出了唯一值的张量。

PyTorch中的torch.unique()函数还有其他参数,可以用于控制返回结果的排序、返回索引等。具体的使用方法可以参考PyTorch官方文档中torch.unique()的说明:torch.unique()

腾讯云相关产品和产品介绍链接地址:

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

领券