PyTorch torch.abs 関数
torch.absは PyTorch においてテンソルの絶対値を計算するための関数です。
関数定義
torch.abs(input, out=None)
使用例
実例
import torch
# 負の数を含むテンソルを作成
x = torch.tensor([-1, -2, 3, -4])
# 絶対値を計算
y = torch.abs(x)
print(y)
# 負の数を含むテンソルを作成
x = torch.tensor([-1, -2, 3, -4])
# 絶対値を計算
y = torch.abs(x)
print(y)
出力結果は次のとおりです:
tensor([1, 2, 3, 4])
その他の拡張
Pytorch torch リファレンスマニュアル