megengine.functional.sign

sign(inp)[源代码]

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

参数

输入张量。

返回

输入张量的符号。

例如:

from megengine import tensor
import megengine.functional as F

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

输出:

[ 1 -1  0]