Python numpy.dot()

numpy.dot() 是 numpy 模块中的一个函数，它用于矩阵乘法或向量乘法运算。

语法

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

其中：

• a，b：要进行矩阵乘法或向量乘法的输入数组。
• out：用于指定输出结果的可选参数。

返回值

numpy.dot() 函数返回两个数组的乘积，结果类型根据两个输入数组的类型进行推断。

示例

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 = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.dot(a, b)

print(c)
# 32

注意事项

• 如果两个输入数组是一维数组，numpy.dot() 函数将计算它们的点积。
• 如果输入数组的维度大于1，numpy.dot() 函数将按照矩阵乘法的规则进行计算。
• 如果输入的两个数组的维度不一致，numpy.dot() 函数会先尝试将它们进行转置，然后进行相乘运算。
• numpy.dot() 函数也可以用于计算多个矩阵乘积，例如 np.dot(a, b, c) 表示计算矩阵 a、b 和 c 的乘积。

总之，numpy.dot() 函数是一个非常实用的函数，可以用于矩阵乘法或向量乘法计算。