PyTorch torch.allclose 関数
Pytorch torch リファレンスマニュアルtorch.allcloseは、PyTorch で2つのテンソルのすべての要素が近いかどうかを検査する関数です。指定された許容範囲内で2つのテンソルが等しいかどうかを判断します。
関数定義
torch.allclose(input, other, rtol=1e-05, atol=1e-08, equal_nan=False)
使用例
実例
import torch
# 2つの近いテンソルを作成
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# すべての要素が近いかどうかを確認
result = torch.allclose(x, y)
print(result)
# 2つの近いテンソルを作成
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([1.0, 2.0, 3.0 + 1e-9])
# すべての要素が近いかどうかを確認
result = torch.allclose(x, y)
print(result)
出力結果は次のとおりです：
True
その他の拡張