PyTorch torch.unique_consecutive 関数
Pytorch torch リファレンスマニュアルtorch.unique_consecutivePyTorch で連続する一意の値を取得するための関数です。テンソル内の隣接する連続する一意の値を返します。
関数定義
torch.unique_consecutive(input, return_inverse=False, return_counts=False, dim=None)
使用例
例
import torch
# テンソルを作成
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# 連続する一意の値を取得
unique_vals = torch.unique_consecutive(x)
print(f"Consecutive unique values: {unique_vals}")
# 件数を返す
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Counts: {counts}")
# テンソルを作成
x = torch.tensor([1, 1, 1, 2, 2, 3, 3, 3, 1, 1])
# 連続する一意の値を取得
unique_vals = torch.unique_consecutive(x)
print(f"Consecutive unique values: {unique_vals}")
# 件数を返す
unique_vals, counts = torch.unique_consecutive(x, return_counts=True)
print(f"Counts: {counts}")
その他の拡張