torch.sign(input, out=None)
说明:符号函数,返回一个新张量,包含输入input张量每个元素的正负(大于0的元素对应1,小于0的元素对应-1,0还是0)
参数:
input(Tensor) -- 输入张量
out(Tensor,可选) -- 输出张量
>>> a = torch.randn(4)
>>> a
tensor([ 0.7734, 0.5677, -0.3896, 1.9878])
>>> torch.sign(a)
tensor([ 1., 1., -1., 1.])