torch.sign()详解

torch.sign()是pytorch中的一个函数，用于计算输入张量的元素的符号。

具体地说，torch.sign()函数会返回一个新的张量，其中的元素是输入张量中对应元素的符号。如果输入张量的元素为正数，则对应位置的输出张量元素为1；如果输入张量的元素为负数，则对应位置的输出张量元素为-1；如果输入张量的元素为0，则对应位置的输出张量元素也为0。

以下是一个示例，

import torch

x = torch.tensor([-2, 0, 3.5])
result = torch.sign(x)

print(result)

"""
输出结果为: tensor([-1.,  0.,  1.])
"""

在这个例子中，输入张量x包含了三个元素：-2、0和3.5。torch.sign(x)返回一个新的张量，其中的元素根据输入张量的符号进行填充。

需要注意的是，torch.sign()函数的输入张量可以是任意形状的张量，它会对张量中的每个元素进行符号计算，并返回相同形状的输出张量。