PyTorch torch.unique 函数
Pytorch torch 参考手册torch.unique 是 PyTorch 中用于获取唯一值的函数。返回张量中的唯一值，并可以选择返回对应的索引。
函数定义
torch.unique(input, sorted=True, return_inverse=False, return_counts=False, dim=None)
使用示例
实例
import torch
# 创建张量
x = torch.tensor([1, 2, 2, 3, 1, 4, 3])
# 获取唯一值
unique_vals = torch.unique(x)
print(f"Unique values: {unique_vals}")
# 同时返回索引
unique_vals, inverse_idx = torch.unique(x, return_inverse=True)
print(f"Inverse indices: {inverse_idx}")
# 同时返回计数
unique_vals, counts = torch.unique(x, return_counts=True)
print(f"Counts: {counts}")
# 创建张量
x = torch.tensor([1, 2, 2, 3, 1, 4, 3])
# 获取唯一值
unique_vals = torch.unique(x)
print(f"Unique values: {unique_vals}")
# 同时返回索引
unique_vals, inverse_idx = torch.unique(x, return_inverse=True)
print(f"Inverse indices: {inverse_idx}")
# 同时返回计数
unique_vals, counts = torch.unique(x, return_counts=True)
print(f"Counts: {counts}")