megengine.Tensor.mean¶
- Tensor.mean(axis=None, keepdims=False)¶
返回在给定的维度
axis上输入张量每一行的平均值。如果
axis是一个包含多个轴的列表，则会在它们上都做规约。如果keepdims是True, 那么输出张量的形状保持和输入张量一致，除了axis指定的维度值将变为 1. 否则axis轴会被压缩（参考squeeze）.- 参数
axis – 做规约的维度或者多个维度
keepdims (
bool) – 输出张量是否需要保留原始数量的维度。
- 返回
输出 Tensor.
实际案例
from megengine import tensor a = tensor([False, True, True, False]) b = tensor([1.0, 2.0, 3.0, 4.0]) print(a.mean().numpy()) print(b.mean().numpy())
输出：
0.5 2.5