megengine.functional.square¶
- square(x)[源代码]¶
元素 \(x^2\) 函数
计算输入 Tensor \(x\) 中每个元素 \(x_i\) 的平方数。
- 参数
x (
Tensor) – 输入 tensor。应该具有 float 数据类型。- 返回类型
- 返回
一个 Tensor ，包含 \(x\) 中每个元素的平方根计算结果。返回的 Tensor 将会使用浮点数据类型，具体见 类型提升规则 。
实际案例
>>> F.square(2) Tensor(4, dtype=int32, device=xpux:0)
Element-wise square:
>>> x = Tensor([1, -2, -3, 4]) >>> F.square(x) Tensor([ 1 4 9 16], dtype=int32, device=xpux:0)