Numpy中二维与三维矩阵相乘(Numpy,2D and 3D matrix multiplication)
发布时间
阅读量:
阅读量
Numpy中二维矩阵与三维矩阵的相乘操作(Numpy, 2D matrix multiplies 3D matrix)
掌握在Numpy中应用广播规则的方法。
========
目标:实现逐元素的矩阵乘法运算
参数设置如下:
A:尺寸为2×3×4的三维数组
B:尺寸为2×4的二维数组
C = AB (此处表示Numpy中的逐元素乘法运算)
========
代码示例如下:
import numpy as np
A = np.ones([2, 3, 4])
B = np.ones([2, 4])
C = A*B[:, None, :]
由于B[:, None, :]的维度变为2×1×4,其中第二个轴的长度为1,因此可以顺利应用广播规则进行计算。
以上即为全部内容。

全部评论 (0)
还没有任何评论哟~
