評價此頁

torch.resolve_neg#

torch.resolve_neg(input) Tensor#

如果 input 的負數位設定為 True,則返回一個帶有實際負數的全新張量;否則返回 input。輸出張量的負數位將始終設定為 False

引數

input (Tensor) – 輸入張量。

示例

>>> x = torch.tensor([-1 + 1j, -2 + 2j, 3 - 3j])
>>> y = x.conj()
>>> z = y.imag
>>> z.is_neg()
True
>>> out = z.resolve_neg()
>>> out
tensor([-1., -2., 3.])
>>> out.is_neg()
False