np.dot()函数
时间: 2023-07-18 11:10:34 浏览: 130
np.dot()函数的用法详解
np.dot()函数是NumPy中的矩阵乘法函数,用于计算两个数组的点积或者矩阵乘法。它可以接受两个参数,分别是要相乘的两个数组。如果两个数组是一维的,则计算的是它们的内积;如果是二维数组(矩阵),则计算的是它们的矩阵乘法;如果是多维数组,则可以使用transpose方法来交换数组的轴,以满足矩阵乘法的要求。例如:
```
import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
# 一维数组的点积
c = np.dot(a, b)
print(c) # 32
x = np.array([[1, 2], [3, 4]])
y = np.array([[5, 6], [7, 8]])
# 矩阵乘法
z = np.dot(x, y)
print(z)
# [[19 22]
# [43 50]]
```
阅读全文