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
本页目录