跳转至

16 NumPy中的矩阵乘法

NumPy中的矩阵乘法(Matrix Multiplication)

矩阵的乘法是一种操作,通过将两个矩阵作为输入,将第一个矩阵的行与第二个矩阵的列相乘,生成一个单独的矩阵。需要注意的是,第一个矩阵的行数应等于第二个矩阵的列数。

Matrix

在Python中,使用NumPy进行矩阵乘法的过程被称为向量化。向量化的主要目标是消除或减少我们明确使用的for循环。通过从程序中减少“for”循环可以加快计算速度。内置的NumPy包用于数组处理和操作。 有三种方法可以执行numpy矩阵乘法。

  • 第一种是使用multiply()函数,该函数执行矩阵的逐元素乘法。

  • 第二种是使用matmul()函数,该函数执行两个数组的矩阵乘积。

  • 最后一种是使用dot()函数,该函数执行两个数组的点积。

示例 1:逐元素矩阵乘法

import numpy as np  

array1=np.array([[1,2,3],[4,5,6],[7,8,9]],ndmin=3)  
array2=np.array([[9,8,7],[6,5,4],[3,2,1]],ndmin=3)  
result=np.multiply(array1,array2)  
result

在上面的代码中

  • 我们导入了别名为np的numpy。

  • 我们使用numpy.array()函数创建了array1和array2,维数为3。

  • 我们创建了一个名为result的变量,并将np.multiply()函数的返回值赋给它。

  • 我们在np.multiply()中传递了array1和array2两个数组。

  • 最后,我们尝试打印result的值。

在输出中,显示了一个三维矩阵,其元素是array1和array2元素的逐元素乘积的结果。 输出:

array([[[ 9, 16, 21],
            [24, 25, 24],
            [21, 16,  9]]])

示例 2:矩阵乘积

import numpy as np  

array1=np.array([[1,2,3],[4,5,6],[7,8,9]],ndmin=3)  
array2=np.array([[9,8,7],[6,5,4],[3,2,1]],ndmin=3)  
result=np.matmul(array1,array2)  
result
输出:
array([[[ 30,  24,  18],
            [ 84,  69,  54],
            [138, 114,  90]]])
在上面的代码中

  • 我们导入了别名为np的numpy。

  • 我们使用numpy.array()函数创建了array1和array2,维数为3。

  • 我们创建了一个名为result的变量,并将np.matmul()函数的返回值赋给它。

  • 我们在np.matmul()中传递了array1和array2两个数组。

  • 最后,我们尝试打印result的值。

在输出中,显示了一个三维矩阵,其元素是array1和array2元素的乘积结果。

示例 3:点积

numpy.dot具有以下规格:

  • 当a和b都是1-D(一维)数组时-> 两个向量的内积(不包含复共轭)

  • 当a和b都是2-D(二维)数组时 -> 矩阵乘法

  • 当a或b为0-D(也称为标量)时 -> 使用numpy.multiply(a, b)或a * b进行乘法。

  • 当a是N-D数组,b是1-D数组时 -> 在a和b的最后一个轴上进行求和乘积。

  • 当a是N-D数组,b是M-D数组,并且M>=2时 -> 在a的最后一个轴和b的倒数第二个轴上进行求和乘积:同时,dot(a, b)[i,j,k,m] = sum(a[i,j,:] * b[k,:,m])

import numpy as np  

array1=np.array([[1,2,3],[4,5,6],[7,8,9]],ndmin=3)  
array2=np.array([[9,8,7],[6,5,4],[3,2,1]],ndmin=3)  
result=np.dot(array1,array2)  
result
在上面的代码中

  • 我们导入了别名为np的numpy。

  • 我们使用numpy.array()函数创建了array1和array2,维数为3。

  • 我们创建了一个名为result的变量,并将np.dot()函数的返回值赋给它。

  • 我们在np.dot()中传递了array1和array2两个数组。

  • 最后,我们尝试打印result的值。

在输出中,显示了一个三维矩阵,其元素是array1和array2元素的点积结果。 输出:

array([[[[ 30,  24,  18]],
            [[ 84,  69,  54]],
            [[138, 114,  90]]]])