大家好,谢谢你们的帮助。我对python很陌生,也没有为下面的任务找到一个有效的替代for循环的方法。
我想分别乘维数A和B的维数(d,n,m)和(d,m)。使用一些术语来帮助理解,A是nxm矩阵的列表,B是R^m中的向量列表。
例如:
A = np.array([[[0,0,0,0,0],[1,1,1,1,1],[2,2,2,2,2]],[[3,3,3,3,3],[4,4,4,4,4],[5,5,5,5,5]]])
B = np.array([[1,2,3,4,5],[5,6,7,8,9]])
我的解决方案使用for循环。
for i in range(2):
print(A[i]*