matmul
是 Open3D 中的一个函数,用于计算两个张量的矩阵乘法。本函数对两个张量进行相乘,并返回结果张量。
matmul(a: open3d.core.Tensor, b: open3d.core.Tensor) -> open3d.core.Tensor
import open3d.core as o3c
import numpy as np
a = o3c.Tensor(np.array([[1,2], [3,4]], dtype=np.float32))
b = o3c.Tensor(np.array([[5,6], [7,8]], dtype=np.float32))
# 计算矩阵乘积
c = o3c.matmul(a,b)
print(c)
# 输出结果: [[19, 22], [43, 50]]
a
和 b
的张量秩不同,则会触发 ValueError
异常。a
和 b
的第二个维度大小不同,则会触发 ValueError
异常。