np.dot
是 NumPy 库中的一个函数,用于计算两个数组的点积。它支持多种类型的输入数组,包括向量和矩阵,并返回它们的点积或矩阵乘积。
对于两个向量(一维数组),np.dot
返回它们的点积,即对应元素的乘积之和。对于矩阵和向量的乘积,或者两个矩阵的乘积,np.dot
执行标准的线性代数矩阵乘法。
下面是一些 np.dot
的基本用法示例:
两个向量的点积
python复制代码
import numpy as np |
|
a = np.array([1, 2, 3]) |
|
b = np.array([4, 5, 6]) |
|
result = np.dot(a, b) |
|
print(result) # 输出:32 |
矩阵与向量的乘积
python复制代码
A = np.array([[1, 2], [3, 4]]) |
|
b = np.array([5, 6]) |
|
result = np.dot(A, b) |
|
print(result) # 输出:[17 39] |
两个矩阵的乘积
python复制代码
A = np.array([[1, 2], [3, 4]]) |
|
B = np.array([[5, 6], [7, 8]]) |
|
result = np.dot(A, B) |
|
print(result) |
|
# 输出: |
|
# [[19 22] |
|
# [43 50]] |
请注意,当计算两个矩阵的乘积时,左侧矩阵的列数必须等于右侧矩阵的行数,否则 NumPy 会抛出一个错误。
np.dot
在进行数组运算时非常高效,因为它利用了底层的优化算法。在处理大型数组或进行复杂的数学运算时,使用 np.dot
通常比手动实现这些运算要快得多。