mindspore.ops.Sign
- class mindspore.ops.Sign[源代码]
符号函数,计算输入Tensor元素的执行符号。注意:在输入类型为float64时,该算子反向计算结果为NaN。
\[sign(x) = \begin{cases} -1, &if\ x < 0 \cr 0, &if\ x = 0 \cr 1, &if\ x > 0\end{cases}\]- 输入:
x (Tensor) - 任意维度输入Tensor。
- 输出:
Tensor,shape和数据类型与 x 相同。
- 异常:
TypeError - 如果 x 不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> x = Tensor(np.array([[2.0, 0.0, -1.0]]), mindspore.float32) >>> sign = ops.Sign() >>> output = sign(x) >>> print(output) [[ 1. 0. -1.]]