megengine.functional.var¶

var(inp, axis=None, keepdims=False)[源代码]¶

计算给定维度 (或坐标轴) 元素的方差

参数

inp (Tensor) – 输入张量。应该有一个数字数据类型。
axis (Union[int, Sequence[int], None]) – 计算方差的维度或坐标轴。默认情况下，方差计算范围是 Tensor 中的全部元素。如果参数是整数数组，方差将按照数组中整数给定的维度进行计算。
keepdims (bool) – 若为 “True”，则指定用来reduce的轴(维度)必须包含在结果中，因此，结果必须与输入tensor兼容(参见:ref:`broadcasting-rule `)。否则，若为 “False”，结果中不得包含用于reduce的轴(维度)。

返回类型

Tensor

返回

如果对 Tensor 中的所有元素进行计算，返回没有维度的 Tensor ，否则返回有维度的 Tensor。所返回 Tensor 的数据类型将由类型提升规则决定。

注解

方差是标准差平方后的均值结果，即 var = mean(x), where x = abs(a - a.mean())**2

实际案例

>>> x = Tensor([[1, 2], [3, 4]])
>>> F.var(x)
Tensor(1.25, device=xpux:0)

>>> x = Tensor([[14, 8, 11, 10], [7, 9, 10, 11], [10, 15, 5, 10]])
>>> F.var(x)
Tensor(6.8333335, device=xpux:0)

megengine.functional.max

megengine.functional.std