Advertisement

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)

还没有任何评论哟~