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