PyTorch torch.unique 関数
Pytorch torch リファレンスマニュアルtorch.uniquePyTorch で一意の値を取得するための関数です。テンソル内の一意の値を返し、オプションで対応するインデックスも返すことができます。
関数定義
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}")
その他の拡張