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