open3d.core.Tensor.mean
函数用于计算张量的平均值。
mean(dim=None, keepdims=False) -> open3d.core.Tensor
dim
(int, tuple, list, optional) - 沿指定维度计算平均值。如果未指定维度,则计算所有元素的平均值。默认值为 None
。keepdims
(bool, optional) - 如果设置为 True
,则保留输出的尺寸大小。否则,缩减输出的维度。默认值为 False
。返回一个新的张量,包含沿指定维度计算的平均值。
以下示例演示如何计算张量的平均值:
import open3d.core as o3c
import numpy as np
# 创建张量
t = o3c.Tensor(np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]))
# 计算所有元素的平均值
mean_all = t.mean()
print(mean_all) # 输出:Tensor[1. 2. 3. 4. 5. 6.]
# 沿行的方向计算平均值
mean_rows = t.mean(dim=0)
print(mean_rows) # 输出:Tensor[2.5 3.5 4.5]
# 沿列的方向计算平均值
mean_cols = t.mean(dim=1)
print(mean_cols) # 输出:Tensor[2. 5.]
IndexError
异常。keepdims
为 True
以保留其尺寸大小。否则,则将其缩减为大小为1的维度。