如何解决2D数组和3D数组的乘法
我有3D A矩阵3x3x5(第三维是5)和2D B矩阵(3x3)。我想将A和B相乘以获得(3x3x5)矩阵。然后将所得3D矩阵的元素求和以创建2D矩阵(3x3)。我该怎么办?解决方法
只需使用*
运算符即可将numpy数组相乘。
import numpy as np
a = np.arange(45).reshape(3,3,5)
b = np.arange(9).reshape(3,3)
c = a * b
print(c) # 3x3x5 array
d = np.sum(c,axis=-1)
print(d)
d
应该是您要找的答案。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。