我有两个不同维数的Numpy数组,如下所示:
arr_a.shape #(2, 10000, 300)
arr_b.shape #(2, 300)
我想像这样计算上面数组的内积
s1 = np.inner(arr_a[0], arr_b[0]) #s1.shape (10000,)
s2 = np.inner(arr_a[1], arr_b[1) #s2.shape (10000,)
有没有更好的方法在一行中做到这一点,而不是单独做两次。
https://stackoverflow.com/questions/66158583
复制