open3d.core.Tensor.sum
是Open3D中的一个函数,用于计算张量的各个维度上的元素总和。
open3d.core.Tensor.sum(dim=None, keepdim=False)
dim
(int, tuple, optional):表示要进行求和的维度,可以指定多个维度,若不指定则按照默认值None对所有维度求和。
keepdim
(bool, optional):表示对求和后的结果是否保持张量原来的维度大小。
open3d.core.Tensor
:返回求和后的张量。
下面的示例将演示如何使用sum
函数对张量中的元素进行求和。
import open3d.core as o3c
import numpy as np
a = np.random.rand(2, 3)
tensor_a = o3c.Tensor(a)
print("原始张量:\n", tensor_a)
print("对所有维度求和后的张量:\n", tensor_a.sum())
print("对第0维度求和后的张量:\n", tensor_a.sum(dim=0))
print("对第1维度求和后的张量:\n", tensor_a.sum(dim=1))
print("对第1维度求和后保持张量原来维度大小的张量:\n", tensor_a.sum(dim=1, keepdim=True))
输出:
原始张量:
[[0.06526229 0.064944 0.69373677]
[0.90736771 0.08021961 0.45985956]]
对所有维度求和后的张量:
1.37138994
对第0维度求和后的张量:
[0.97263001 0.14516361 1.15359633]
对第1维度求和后的张量:
[0.82494306 1.44744689]
对第1维度求和后保持张量原来维度大小的张量:
[[1.82494324]
[2.4474473 ]]
ValueError
(“Dim out of range”):若指定的要进行求和的维度超出张量的维度范围,则会抛出该异常。