PyTorch torch.sign 関数
Pytorch torch リファレンスマニュアルtorch.signPyTorch において要素ごとに符号を返す関数です。-1（負の数）、0（ゼロ）、または 1（正の数）を返します。
関数定義
torch.sign(input, out=None)
使用例
実例
import torch
# 要素ごとに符号を返す
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sign(x)
print(result)
# 要素ごとに符号を返す
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sign(x)
print(result)
その他の拡張