【NumPy】深入理解NumPy的dot函数：矩阵乘法与点积运算详解

深入理解NumPy的dot函数：矩阵乘法与点积运算详解。numpy.dot 函数用于计算两个数组的点积。根据输入数组的维度不同，可用于不同类型的计算：1. 一维数组：计算两个一维数组的内积（即点积）。2. 二维数组：进行矩阵乘法。3. 多维数组：按照传统的矩阵乘法规则计算其对应的相乘。

---

🧑 博主简介：
现任阿里巴巴嵌入式技术专家，15年工作经验，深耕嵌入式+人工智能领域，精通嵌入式领域开发、技术管理、简历招聘面试。CSDN优质创作者，提供产品测评、学习辅导、简历面试辅导、毕设辅导、项目开发、C/C++/Java/Python/Linux/AI等方面的服务，如有需要请站内私信或者联系任意文章底部的的VX名片（ID：gylzbk）

💬 博主粉丝群介绍：① 群内高中生、本科生、研究生、博士生遍布，可互相学习，交流困惑。② 热榜top10的常客也在群里，也有数不清的万粉大佬，可以交流写作技巧，上榜经验，涨粉秘籍。③ 群内也有职场精英，大厂大佬，可交流技术、面试、找工作的经验。④ 进群免费赠送写作秘籍一份，助你由写作小白晋升为创作大佬。⑤ 进群赠送CSDN评论防封脚本，送真活跃粉丝，助你提升文章热度。有兴趣的加文末联系方式，备注自己的CSDN昵称，拉你进群，互相学习共同进步。

1. NumPy库介绍

NumPy（Numerical Python）是Python编程语言的一个库，提供了对大型、多维数组和矩阵的支持，此外还提供了许多数学函数库用于对数组进行操作。NumPy是科学计算的基础工具之一，其高效处理和构建多维数组的能力使其在数据科学、机器学习和科学计算中被广泛使用。

NumPy数组对象（ndarray）比Python标准列表有几个优势：其大小固定、元素类型相同、内存布局紧凑且支持矢量化操作。这一系列特性使得NumPy成为有需要进行数值运算和处理真实世界大数据的高性能计算程序员的首选工具。

2. dot函数介绍

numpy.dot 函数用于计算两个数组的点积。根据输入数组的维度不同，可用于不同类型的计算：

1. 一维数组：计算两个一维数组的内积（即点积）。
2. 二维数组：进行矩阵乘法。
3. 多维数组：按照传统的矩阵乘法规则计算其对应的相乘。

2.1 函数定义

numpy.dot(a, b, out=None)

参数说明

• a：第一个输入数组。
• b：第二个输入数组。
• out：一个存储结果的输出数组，可选项。

返回值

返回两个输入数组的点积。如果提供了out参数，结果会存储在out参数中并返回。

接下来，我们将通过一系列示例展示numpy.dot的不同用法。

3. 示例代码

在接下来的部分中，我们将演示numpy.dot 在一维数组、二维数组和多维数组上的应用。

3.1 一维数组点积

假设我们有两个一维数组x和y，我们可以通过numpy.dot 计算它们的点积。

import numpy as np

# 创建两个一维数组
x = np.array([1, 2, 3])
y = np.array([4, 5, 6])

# 计算点积
dot_product = np.dot(x, y)
print(dot_product)

输出如下：

32

计算过程为：(14 + 25 + 3*6 = 4 + 10 + 18 = 32)。

3.2 二维数组（矩阵）乘法

当我们处理二维数组时，numpy.dot 用于进行矩阵乘法运算。

import numpy as np

# 创建两个二维数组
A = np.array([[1, 2], 
              [3, 4]])
B = np.array([[5, 6], 
              [7, 8]])

# 计算矩阵乘法
product = np.dot(A, B)
print(product)

输出如下：

[[19 22]
 [43 50]]

计算过程为：

[[1*5 + 2*7, 1*6 + 2*8], 
 [3*5 + 4*7, 3*6 + 4*8]]

即：

[[19, 22], 
 [43, 50]]

3.3 多维数组点积

对于多维数组，numpy.dot 按照传统矩阵乘法规则进行相乘。

import numpy as np

# 创建两个三维数组
A = np.array([[[1, 2], 
               [3, 4]], 
              
              [[5, 6], 
               [7, 8]]])
B = np.array([[[1, 0], 
               [0, 1]], 
              
              [[1, 1], 
               [1, 1]]])

# 计算多维数组乘法
product = np.dot(A, B)
print(product)

输出如下：

[[[ 1  2]
  [ 3  4]
  [ 3  3]
  [ 7  7]]

 [[ 5  6]
  [ 7  8]
  [11 11]
  [15 15]]]

这里展示的是两个三维张量的点积计算，点积操作根据矩阵乘法规则在不同维度上执行。

3.4 使用输出数组

我们还可以通过提供out参数将结果存储在已存在的数组中。

import numpy as np

# 创建两个数组
x = np.array([1, 2, 3])
y = np.array([4, 5, 6])

# 创建存储结果的数组
out_array = np.zeros(1)

# 计算点积并存储结果
np.dot(x, y, out=out_array)
print(out_array)

输出如下：

[32.]

在这种情况下，计算结果32存储在out_array中。

3.5 实际应用：图像处理中的点积计算

在图像处理和计算机视觉中，点积计算常用于卷积操作、图像滤波等。例如，我们想要对一个2D灰度图像应用一个简单的3x3卷积核：

import numpy as np
from scipy.signal import convolve2d

# 创建一个灰度图像（6x6数组）
image = np.array([[1, 2, 3, 4, 5, 6],
                  [7, 8, 9, 10, 11, 12],
                  [13, 14, 15, 16, 17, 18],
                  [19, 20, 21, 22, 23, 24],
                  [25, 26, 27, 28, 29, 30],
                  [31, 32, 33, 34, 35, 36]])

# 定义一个3x3卷积核
kernel = np.array([[1, 0, -1],
                   [1, 0, -1],
                   [1, 0, -1]])

# 使用卷积计算
convolved_image = convolve2d(image, kernel, mode='valid')
print(convolved_image)

输出如下：

[[ -6  -6  -6  -6]
 [-18 -18 -18 -18]
 [-30 -30 -30 -30]
 [-42 -42 -42 -42]]

这个示例展示了如何通过卷积计算边缘检测。卷积运算实质上是一个滑动窗口的点积计算，它在卷积核覆盖的每个局部区域执行点积操作。

4. 总结

NumPy作为一个功能丰富的库，为科学计算和数据处理提供了极大的便利。本文重点介绍了NumPy中的dot函数，这是一个用于计算点积和矩阵乘法的重要函数。通过不同维度的数组示例，我们展示了numpy.dot计算一维数组的点积、二维数组的矩阵乘法和多维数组的点积。此外，我们还演示了如何指定输出数组以及在图像处理中的实际应用。

从这篇文章中，我们了解到：

1. numpy.dot 是一个多用途的函数，可以根据输入数组的维度不同来处理点积和矩阵乘法。
2. 无论是用于简单的向量计算、矩阵操作还是复杂的多维张量乘法，numpy.dot 均表现出色。
3. 在实际应用中，通过点积计算，我们可以实现许多重要操作，如卷积、图像滤波等。

NumPy不仅仅局限于dot函数，它还提供了一系列强大而高效的数值计算工具。掌握这些工具可以大大提升我们在科学计算、数据分析等领域的工作效率和准确性。如果你对NumPy及其功能有更多的兴趣，建议进一步学习和探索更多的相关内容。