mindspore.ops.Sign

class mindspore.ops.Sign[源代码]

符号函数,计算输入Tensor元素的执行符号。

\[sign(x) = \begin{cases} -1, &if\ x < 0 \cr 0, &if\ x = 0 \cr 1, &if\ x > 0\end{cases}\]

输入:

  • x (Tensor) - Sign的输入,shape: \((N, *)\) ,其中 \(*\) 表示任意数量的附加维度。

输出:

Tensor,shape和数据类型与 x 相同。

异常:

  • TypeError - 如果 x 不是Tensor。

支持平台:

Ascend CPU GPU

样例:

>>> x = Tensor(np.array([[2.0, 0.0, -1.0]]), mindspore.float32)
 >>> sign = ops.Sign()
 >>> output = sign(x)
 >>> print(output)
 [[ 1.  0. -1.]]