torch.signbit

torch.signbit(input, *, out=None) → Tensor

判断input中的每个元素是否设置了符号位。

参数

input (Tensor) – 需要输入的张量。

关键字参数

out (Tensor, 可选) – 指定输出张量。

示例：

>>> a = torch.tensor([0.7, -1.2, 0., 2.3])
>>> torch.signbit(a)
tensor([ False, True,  False,  False])
>>> a = torch.tensor([-0.0, 0.0])
>>> torch.signbit(a)
tensor([ True,  False])

注意

signbit 能处理带符号的零，因此对于负零 (-0)，它会返回 True。