在numpy中,我們可以使用numpy.dot()
函數來計算兩個數組的點積。點積是兩個數組的對應元素相乘后再相加的結果。下面是幾個使用numpy.dot()
函數的技巧:
import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
result = np.dot(a, b)
print(result) # Output: 32
import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
result = np.dot(a, b)
print(result)
# Output:
# [[19 22]
# [43 50]]
import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.array([5, 6])
result = np.dot(a, b)
print(result) # Output: [17 39]
注意,點積的計算規則是第一個數組的最后一個維度與第二個數組的倒數第二個維度相等。因此,要確保數組的維度匹配才能正確計算點積。