pts = np.array([
[1, 1, 1],
[2, 2, 2],
[3, 3, 3],
[4, 4, 4],
[5, 5, 5],
])
我想要将这些点中的每一个都乘以一个变换矩阵:
pts[0] = np.dot(transform_matrix, pts[0])
pts[1] = np.dot(transform_matrix, pts[1])
…
pts[n] = np.dot(transform_matrix, pts[n])
我该如何高效地完成这个任务?
einsum
的技巧很好,谢谢。 - David Wolever