Python中使用愛因斯坦求和約定進行向量內積
要使用愛因斯坦求和約定計算向量的內積,請在 Python 中使用 numpy.einsum() 方法。第一個引數是下標。它指定作為逗號分隔的下標標籤列表的求和下標。第二個引數是運算元。這些是操作的陣列。
einsum() 方法對運算元評估愛因斯坦求和約定。使用愛因斯坦求和約定,許多常見的多分量線性代數陣列運算可以用簡單的方式表示。在隱式模式下,einsum 計算這些值。
在顯式模式下,einsum 透過停用或強制對指定下標標籤進行求和,提供了進一步的靈活性來計算可能不被認為是經典愛因斯坦求和運算的其他陣列運算。
步驟
首先,匯入所需的庫:
import numpy as np
使用 arange() 和 reshape() 方法建立一個 numpy 陣列:
arr = np.arange(4)
顯示陣列:
print("Our Array...\n",arr)
檢查維度:
print("\nDimensions of our Array...\n",arr.ndim)
獲取資料型別:
print("\nDatatype of our Array object...\n",arr.dtype)
獲取形狀:
print("\nShape of our Array object...\n",arr.shape)
要使用愛因斯坦求和約定計算向量的內積,請使用 numpy.einsum() 方法:
print("\nResult (inner product)...\n",np.einsum('i,i', arr, arr))
示例
import numpy as np # Creating a numpy array using the arange() and reshape() method arr = np.arange(4) # Display the array print("Our Array...\n",arr) # Check the Dimensions print("\nDimensions of our Array...\n",arr.ndim) # Get the Datatype print("\nDatatype of our Array object...\n",arr.dtype) # Get the Shape print("\nShape of our Array object...\n",arr.shape) # To compute inner product of vectors with Einstein summation convention, use the numpy.einsum() method in Python. print("\nResult (inner product)...\n",np.einsum('i,i', arr, arr))
輸出
Our Array... [0 1 2 3] Dimensions of our Array... 1 Datatype of our Array object... int64 Shape of our Array object... (4,) Result (inner product)... 14
廣告