獲取一維和二維陣列的內積


要獲取兩個陣列的內積,在 Python 中使用 numpy.inner() 方法。向量的一般內積是一維陣列,高維度中是對最後軸的和積。引數是 a 和 b,兩個向量。如果 a 和 b 為非標量,則它們的最後一個維度必須匹配。

步驟

首先,匯入所需的庫 −

import numpy as np

使用 array() 方法建立兩個 numpy 一維陣列 −

arr1 = np.arange(2).reshape((1,1,2))
arr2 = np.arange(6).reshape((3,2))

顯示陣列 −

print("Array1...\n",arr1)
print("\nArray2...\n",arr2)

檢查兩個陣列的維度 −

print("\nDimensions of Array1...\n",arr1.ndim)
print("\nDimensions of Array2...\n",arr2.ndim)

檢查兩個陣列的形狀 −

print("\nShape of Array1...\n",arr1.shape)
print("\nShape of Array2...\n",arr2.shape)

要獲取兩個陣列的內積,請使用 numpy.inner() 方法 −

print("\nResult (Inner Product)...\n",np.inner(arr1, arr2))

示例

import numpy as np

# Creating two numpy One-Dimensional array using the array() method
arr1 = np.arange(2).reshape((1,1,2))
arr2 = np.arange(6).reshape((3,2))

# Display the arrays
print("Array1...\n",arr1)
print("\nArray2...\n",arr2)

# Check the Dimensions of both the arrays
print("\nDimensions of Array1...\n",arr1.ndim)
print("\nDimensions of Array2...\n",arr2.ndim)

# Check the Shape of both the arrays
print("\nShape of Array1...\n",arr1.shape)
print("\nShape of Array2...\n",arr2.shape)

# To get the Inner product of two arrays, use the numpy.inner() method in Python
# Ordinary inner product of vectors for 1-D arrays, in higher dimensions a sum product over the last axes.
print("\nResult (Inner Product)...\n",np.inner(arr1, arr2))

輸出

Array1...
[[[0 1]]]

Array2...
[[0 1]
[2 3]
[4 5]]

Dimensions of Array1...
3

Dimensions of Array2...
2

Shape of Array1...
(1, 1, 2)

Shape of Array2...
(3, 2)

Result (Inner Product)...
[[[1 3 5]]]

更新於: 2022 年 3 月 2 日

262 瀏覽量

開啟你的 職業生涯

完成課程即可獲得認證

開始學習
廣告