import numpy
A = numpy.array([
[0,1,1],
[2,2,0],
[3,0,3]
])
B = numpy.array([
[1,1,1],
[2,2,2],
[3,2,9],
[4,4,4],
[5,9,5]
])
Run Code Online (Sandbox Code Playgroud)
A尺寸:N * N(3*3)
B尺寸:K*N(5*3)
预期结果为: C = [ A * B[0], A * B[1], A * B[2], A * B[3], A * B[4]] (C 的维度也是 5* 3)
我是 numpy 的新手,不确定如何在不使用 for 循环的情况下执行此操作。
谢谢!