API 相关说明
返回一个新的张量,该张量表明每个元素的符号。
inp (Tensor) – Tensor:
Tensor
输入张量的符号。
实际案例
from megengine import tensor import megengine.functional as F x = tensor([1, -1, 0]) print(F.sign(x).numpy())
输出:
[ 1 -1 0]
上一页
megengine.functional.abs
下一页
megengine.functional.pow