快捷方式

torch.sgn

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

此函数是 torch.sign() 对复数张量的扩展。它计算一个新张量,其元素与 input 中对应元素具有相同的角度和值为 1 的绝对值(即幅度),对于非复数张量,它等价于 torch.sign()。

outi={0inputi==0inputiinputiotherwise\text{out}_{i} = \begin{cases} 0 & |\text{{input}}_i| == 0 \\ \frac{{\text{{input}}_i}}{|{\text{{input}}_i}|} & \text{otherwise} \end{cases}
参数

input (张量) – 输入张量。

关键字参数

out (张量, 可选) – 输出张量。

示例

>>> 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])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源