sgn¶
对于复数 Tensor,此函数返回一个新的 Tensor,其元素与 input 元素的角度相同且绝对值为 1。
对于实数 Tensor,对输入参数 x
中每个元素进行正负判断,并且输出正负判断值:1 代表正,-1 代表负,0 代表零。
参数¶
x (Tensor) – 输入 Tensor,数据类型为 float16、float32、float64、complex64 或 complex128。
name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 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.6000000238418579+0.800000011920929j),
(0.2800000011920929-0.9599999785423279j),
0j ,
(0.4472135901451111+0.8944271802902222j)],
[ (0.6000000238418579+0.800000011920929j),
(1+0j) ,
0j ,
(-1+0j) ]])