Python中numpy.dot的用法是什么

小億
93
2024-01-12 13:46:34

numpy.dot(a, b)函數(shù)用于計(jì)算兩個(gè)數(shù)組的點(diǎn)積（內(nèi)積）。

參數(shù)a和b可以是一維或二維數(shù)組。如果兩個(gè)參數(shù)都是一維數(shù)組，那么函數(shù)將計(jì)算它們的內(nèi)積。如果其中一個(gè)參數(shù)是二維數(shù)組，那么函數(shù)將返回維度較高的數(shù)組的行與維度較低的數(shù)組的列的乘積之和。

示例：

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]]

在多維數(shù)組的情況下，numpy.dot函數(shù)將按照矩陣乘法的規(guī)則進(jìn)行計(jì)算。如果兩個(gè)數(shù)組是一維數(shù)組，那么函數(shù)將計(jì)算它們的點(diǎn)積（內(nèi)積）。

0 踩