PyTorch torch.isin 関数
Pytorch torch リファレンスマニュアルtorch.isintorch.isin は、PyTorch において要素が別のテンソル内に存在するかどうかを確認するための関数です。最初のテンソルの各要素が2番目のテンソルに含まれているかどうかを示すブール型テンソルを返します。
関数定義
torch.isin(elements, test_elements, assume_unique=False, invert=False)
使用例
実例
import torch
# 2つのテンソルを作成
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# 要素がtest_elements内に存在するかどうかを確認
result = torch.isin(elements, test_elements)
print(result)
# 2つのテンソルを作成
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# 要素がtest_elements内に存在するかどうかを確認
result = torch.isin(elements, test_elements)
print(result)
出力結果は次のとおりです：
tensor([False, True, False, True])
その他の拡張