PyTorch torch.not_equal 関数
torch.not_equalPyTorchにおいて、要素ごとに等しくないことを比較するための関数です。最初のテンソルの要素が2番目のテンソルの対応する要素と等しくない位置がTrueとなるテンソルを返します。
関数定義
torch.not_equal(input, other)
使用例
例
import torch
# 2つのテンソルを作成
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 4, 3])
# 要素ごとに等しくないことを比較
result = torch.not_equal(x, y)
print(result)
# 2つのテンソルを作成
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 4, 3])
# 要素ごとに等しくないことを比較
result = torch.not_equal(x, y)
print(result)
出力結果は:
tensor([False, True, False])
その他の拡張
Pytorch torch リファレンスマニュアル