megengine.functional.sign

sign(x)[源代码]

元素求 sign

返回输入 Tensor \(x\) 中每个元素 \(x_i\) 的数字符号值。

参数

inp – 输入张量。 应该有一个数字数据类型。

返回

一个 Tensor ,其中的值为元素 \(x\) 的计算结果,返回的 array 保持和 \(x\) 相同的数据类型。

实际案例

元素求 sign:

>>> x = Tensor([1, -1, 0])
>>> F.sign(x)
Tensor([ 1 -1  0], dtype=int32, device=xpux:0)