PyTorch torch.equal 関数
Pytorch torch リファレンスマニュアルtorch.equalPyTorch で2つのテンソルが等しいかどうかを確認するための関数です。2つのテンソルが同じ形状と要素を持つ場合、True を返します。
関数定義
torch.equal(input, other)
使用例
実例
import torch
# 2つの同じテンソルを作成
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# 等しいかどうかを確認
result = torch.equal(x, y)
print(result)
# 異なるテンソルを作成
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
# 2つの同じテンソルを作成
x = torch.tensor([1, 2, 3])
y = torch.tensor([1, 2, 3])
# 等しいかどうかを確認
result = torch.equal(x, y)
print(result)
# 異なるテンソルを作成
a = torch.tensor([1, 2, 4])
b = torch.tensor([1, 2, 3])
result2 = torch.equal(a, b)
print(result2)
出力結果は次のとおりです：
True False
その他の拡張