PyTorch torch.isclose 関数
Pytorch torch リファレンスマニュアルtorch.iscloseは PyTorch において、2つのテンソルが近いかどうかを確認するための関数です。指定された許容範囲内で、2つのテンソルの対応する要素が近いかどうかを示すブール型テンソルを返します。
関数定義
torch.isclose(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.isclose(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.isclose(x, y)
print(result)
出力結果は次のとおりです：
tensor([True, True, True])
その他の拡張