使用 Python 中的爱因斯坦求和约定计算向量内积

pythonnumpyserver side programmingprogramming更新于 2024/5/14 14:04:17

要使用爱因斯坦求和约定计算向量内积,请使用 Python 中的 numpy.einsum() 方法。第一个参数是下标。它将求和的下标指定为以逗号分隔的下标标签列表。第二个参数是操作数。这些是用于操作的数组。

einsum() 方法评估操作数上的爱因斯坦求和约定。使用爱因斯坦求和约定,许多常见的多维线性代数数组运算可以以简单的方式表示。在隐式模式下,einsum 计算这些值。

在显式模式下,einsum 通过禁用或强制对指定下标标签求和,提供了进一步的灵活性来计算可能不被视为经典爱因斯坦求和运算的其他数组运算。

步骤

首先,导入所需的库 −

import numpy as np

使用 arange() 和 reshape() 方法创建 numpy 数组 −

arr = np.arange(4)

显示数组 −

print("我们的数组...\n",arr)

检查维度 −

print("\n数组的维度...\n",arr.ndim)

获取数据类型 −

print("\n我们的数组对象的数据类型...\n",arr.dtype)

获取形状 −

print("\n我们的数组对象的形状...\n",arr.shape)

要使用爱因斯坦求和约定计算向量的内积,请使用 numpy.einsum() 方法 −

print("\nResult (inner product)...\n",np.einsum('i,i', arr, arr))

示例

import numpy as np

# 使用 arange() 和 reshape() 方法创建 numpy 数组
arr = np.arange(4)

# 显示数组
print("我们的数组...\n",arr)

# 检查维度
print("\n数组的维度...\n",arr.ndim)

# 获取数据类型
print("\n我们的数组对象的数据类型...\n",arr.dtype)

# 获取形状
print("\n我们的数组对象的形状...\n",arr.shape)

# 要使用爱因斯坦求和约定计算向量的内积,请使用 Python 中的 numpy.einsum() 方法。
print("\nResult (inner product)...\n",np.einsum('i,i', arr, arr))

输出

我们的数组...
[0 1 2 3]

我们的数组的维度...
1

我们的数组对象的数据类型...
int64

我们的数组对象的形状...
(4,)

Result (inner product)...
14

相关文章


有用资源