PyTorch torch.sign 函数
Pytorch torch 参考手册torch.sign 是 PyTorch 中用于逐元素返回符号的函数。返回 -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)