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