PyTorch torch.nextafter 関数
torch.nextafterPyTorch において、要素ごとに次の表現可能な浮動小数点数を返すために使用される関数です。
関数定義
torch.nextafter(input, other, out=None)
使用例
実例
import torch
# 次の表現可能な浮動小数点数を返す
x = torch.tensor([1.0, 2.0])
y = torch.tensor([2.0, 2.0])
result = torch.nextafter(x, y)
print(result)
# 次の表現可能な浮動小数点数を返す
x = torch.tensor([1.0, 2.0])
y = torch.tensor([2.0, 2.0])
result = torch.nextafter(x, y)
print(result)
その他の拡張
Pytorch torch リファレンスマニュアル