megengine.functional.sign

sign(inp)[源代码]

返回一个新的张量,该张量表明每个元素的符号。

参数

inp (Tensor) – Tensor:

返回

输入张量的符号。

实际案例

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