Fel*_*lix 2 python numpy matrix scipy
我想得到dot productN个向量对(a_vec [i,:],b_vec [i,:]).
dot product
a_vec具有形状[N, 3],bvec具有相同的形状(N个3D矢量).
a_vec
[N, 3]
bvec
我知道它可以通过numpy.dot功能在循环中轻松完成.但不能以某种方式更简单,更快地完成吗?
numpy.dot
use*_*ica 6
np.einsum 会这样做:
np.einsum
np.einsum('ij,ij->i', a_vec, b_vec)
归档时间:
10 年,5 月 前
查看次数:
86 次
最近记录: