megengine.functional.var

var(inp, axis=None, keepdims=False)

返回在给定轴 axis 上 inp 张量每一行的方差。如果给定的轴是一个维度列表，则规约所有维度。

参数

• inp (Tensor) – 输入张量。

• axis (Union[int, Sequence[int], None]) – 要进行规约的维度。如果设置为 None，则所有的维度都将规约。默认：None

• keepdims (bool) – 输出张量是否保留了轴 axis 。默认：False

返回类型

Tensor

返回

输出张量。

例如：

import numpy as np
from megengine import tensor
import megengine.functional as F

data = tensor(np.arange(1, 7, dtype=np.float32).reshape(2, 3))
out = F.var(data)
print(out.numpy().round(decimals=4))

输出：

2.9167

megengine.functional.svd megengine.functional.std