for文を高速化したい

Question

16次元のベクトルを1152個用意してた1次元配列v1[16*1152]を用意します。
この1152個のベクトルと16次元のベクトルv2[16]との内積をとった結果をb[1152]に保存したいとき,
for文で書くと時間がかかるため, 改善したいです。

for文で書くと, 1152回参照しなければならないので, numpyを使うことで改善できるのかどうか,
教えてください。
よろしくお願いします。

import numpy as np
v1 = np.arange(16*1152)
v2 = np.arange(16)
b = np.zeros(1152)

for i in range(1152):
    b[i] += np.dot(v1[i:i+16],v2)

print(b)

Answer 1

(1152,16) に reshape してから内積をとればいいのではないでしょうか

import numpy as np

v1 = np.arange(16 * 1152)
v2 = np.arange(16)
b = v1.reshape(-1, 16) @ v2
print(b.shape)

Comments

[redcatML]

reshapeは思いつきませんでした。
回答ありがとうございます。