Python PyTorch resolve_neg用法及代码示例

本文简要介绍python语言中 torch.resolve_neg 的用法。

用法:

torch.resolve_neg(input) → Tensor

如果 input 的负位设置为 True ，则返回具有物化否定的新张量，否则返回 input 。输出张量的负位始终设置为 False 。 :param input: 输入张量。 ：类型输入：张量

例子：

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