在 Python 中获取一维和二维数组的内积

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

要获取两个数组的内积,请使用 Python 中的 numpy.inner() 方法。一维数组的普通向量内积,在高维中是最后几个轴的和积。参数是 1 和 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

# 使用 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)

#要获取两个数组的内积,请使用 Python 中的 numpy.inner() 方法
# 一维数组的普通向量内积,在高维中为最后几个轴的和积。
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]]]

相关文章


有用资源