np.dot作用和用法
时间: 2023-10-30 19:55:22 浏览: 70
np.dot()函数的用法详解
np.dot是NumPy库中的一个函数,用于计算两个数组的点积(即矩阵乘法),可以用于矩阵乘法、向量内积等计算。其用法如下:
np.dot(a, b, out=None)
其中,a和b是要计算点积的两个数组,out是可选参数,用于指定输出的数组,如果不指定,则会自动创建一个新的数组来存储结果。
使用示例:
import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
c = np.dot(a, b)
print(c)
# 输出:
# [[19 22]
# [43 50]]
上述代码中,a和b是两个2x2的数组,使用np.dot计算它们的点积,结果存储在变量c中,最终输出了一个2x2的数组。
阅读全文