megengine.functional.sign

sign(inp)[源代码]

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

参数

inp (Tensor) – Tensor:

返回

输入张量的符号。

实际案例

from megengine import tensor
import megengine.functional as F

x = tensor([1, -1, 0])
print(F.sign(x).numpy())

输出:

[ 1 -1  0]