megengine.functional.square

square(x)[源代码]

返回一个包含输入张量中各元素平方的新的张量

参数

inp – 输入张量。

返回

计算得到的张量。

例如:

import numpy as np
import megengine as mge
import megengine.functional as F

data = mge.tensor(np.arange(0, 6, dtype=np.float32).reshape(2, 3))
out = F.square(data)
print(out.numpy().round(decimals=4))

输出:

[[ 0.  1.  4.]
 [ 9. 16. 25.]]