温馨提示×

Python中numpy.dot的用法是什么

小亿
86
2024-01-12 13:46:34
栏目: 编程语言

numpy.dot(a, b)函数用于计算两个数组的点积(内积)。

参数ab可以是一维或二维数组。如果两个参数都是一维数组,那么函数将计算它们的内积。如果其中一个参数是二维数组,那么函数将返回维度较高的数组的行与维度较低的数组的列的乘积之和。

示例:

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
print(np.dot(a, b))  # 输出: 32

c = np.array([[1, 2], [3, 4]])
d = np.array([[5, 6], [7, 8]])
print(np.dot(c, d))  # 输出: [[19 22]
                       [43 50]]

在多维数组的情况下,numpy.dot函数将按照矩阵乘法的规则进行计算。如果两个数组是一维数组,那么函数将计算它们的点积(内积)。

0