sgn

paddle. sgn ( x, name=None, *, out=None ) [源代码] 

对于复数 Tensor，此函数返回一个新的 Tensor，其元素与 input 元素的角度相同且绝对值为 1。

对于实数 Tensor，对输入参数 x 中每个元素进行正负判断，并且输出正负判断值：1 代表正，-1 代表负，0 代表零。

参数

x (Tensor) – 输入 Tensor，数据类型为 float16、float32、float64、complex64 或 complex128。别名 input。

name (str，可选) - 具体用法请参见 api_guide_Name，一般无需设置，默认值为 None。

关键字参数

out (Tensor，可选) - 输出 Tensor，若不为 None，计算结果将保存在该 Tensor 中，默认值为 None。

返回

Tensor，输出正负号或复数的单位向量，数据的 shape 大小及数据类型和输入 x 一致。

代码示例

          >>> import paddle

>>> x = paddle.to_tensor(
...     [
...         [3 + 4j, 7 - 24j, 0, 1 + 2j],
...         [6 + 8j, 3, 0, -2],
...     ]
... )
>>> paddle.sgn(x)
Tensor(shape=[2, 4], dtype=complex64, place=Place(cpu), stop_gradient=True,
[[ (0.60000002+0.80000001j),  (0.28000000-0.95999998j),
   (0.00000000+0.00000000j),  (0.44721359+0.89442718j)],
 [ (0.60000002+0.80000001j),  (1.00000000+0.00000000j),
   (0.00000000+0.00000000j), (-1.00000000+0.00000000j)]])

         

使用本API的教程文档¶

PyTorch 写法