torch.signbit

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

測試 input 的每個元素是否設定了符號位元。

參數

input ( Tensor) – 輸入張量。

關鍵字參數

out ( Tensor, optional) – 輸出張量 (可選)。

範例

>>> 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。