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)