Python PyTorch sgn用法及代碼示例

本文簡要介紹python語言中 torch.sgn 的用法。

用法:

torch.sgn(input, *, out=None) → Tensor

參數：

input(Tensor) -輸入張量。

關鍵字參數：

out(Tensor,可選的) -輸出張量。

此函數是 torch.sign() 對複張量的擴展。它計算一個新張量，其元素與input 的對應元素具有相同的角度，對於複數張量，其絕對值(即幅度)為 1，對於非複數張量，等效於 torch.sign()。

\text{out}_{i} = \begin{cases} 0 & |\text{{input}}_i| == 0 \\ \frac{{\text{{input}}_i}}{|{\text{{input}}_i}|} & \text{otherwise} \end{cases}

例子：

>>> t = torch.tensor([3+4j, 7-24j, 0, 1+2j])
>>> t.sgn()
tensor([0.6000+0.8000j, 0.2800-0.9600j, 0.0000+0.0000j, 0.4472+0.8944j])