跳转至

计算两个多维 NumPy 阵列的 Kronecker 积

原文:https://www . geeksforgeeks . org/compute-the-kronecker-product-of-two-multi dimension-numpy-arrays/

给定一个 m×n 矩阵 a 和一个 p×q 矩阵 b,它们的 Kronecker 积是一个⊗ B,也叫它们的矩阵直积,是一个(mp) X (nq)矩阵。

A = | (a00)  (a01) |
    | (a10)  (a11) |

B = | (b00)  (b01) |
    | (b10)  (b11) |

A ⊗ B = | (a00)*(b00)  (a00)*(b01)  (a01)*(b00)  (a01)*(b00) |
        | (a00)*(b01)  (a00)*(b11)  (a01)*(b01)  (a01)*(b11) | 
        | (a10)*(b00)  (a10)*(b01)  (a11)*(b00)  (a11)*(b01) |
        | (a10)*(b10)  (a10)*(b11)  (a11)*(b10)  (a11)*(b11) |

可以使用 NumPy 模块中的 kron() 方法计算两个给定多维数组的 Kronecker 积。kron()方法将两个数组作为参数,并返回这两个数组的 Kronecker 积。

语法:

numpy.kron(array1, array2)

下面是一些程序,描述了 kron()方法在计算两个数组的 Kronecker 积时的实现:

例 1:

蟒 3

# Importing required modules
import numpy

# Creating arrays
array1 = numpy.array([[1, 2], [3, 4]])
print('Array1:\n', array1)

array2 = numpy.array([[5, 6], [7, 8]])
print('\nArray2:\n', array2)

# Computing the Kronecker Product
kroneckerProduct = numpy.kron(array1, array2)
print('\nArray1 ⊗ Array2:')
print(kroneckerProduct)

输出:

Array1:
 [[1 2]
 [3 4]]

Array2:
 [[5 6]
 [7 8]]

Array1 ⊗ Array2:
[[ 5  6 10 12]
 [ 7  8 14 16]
 [15 18 20 24]
 [21 24 28 32]]

例 2:

蟒 3

# Importing required modules
import numpy

# Creating arrays
array1 = numpy.array([[1, 2, 3]])
print('Array1:\n', array1)

array2 = numpy.array([[3, 2, 1]])
print('\nArray2:\n', array2)

# Computing the Kronecker Product
kroneckerProduct = numpy.kron(array1, array2)
print('\nArray1 ⊗ Array2:')
print(kroneckerProduct)

输出:

Array1:
 [[1 2 3]]

Array2:
 [[3 2 1]]

Array1 ⊗ Array2:
[[3 2 1 6 4 2 9 6 3]]

例 3:

蟒 3

# Importing required modules
import numpy

# Creating arrays
array1 = numpy.array([[1, 2, 3], [4, 5, 6]])
print('Array1:\n', array1)

array2 = numpy.array([[1, 2], [3, 4], [5, 6]])
print('\nArray2:\n', array2)

# Computing the Kronecker Product
kroneckerProduct = numpy.kron(array1, array2)
print('\nArray1 ⊗ Array2:')
print(kroneckerProduct)

输出:

Array1:
 [[1 2 3]
 [4 5 6]]

Array2:
 [[1 2]
 [3 4]
 [5 6]]

Array1 ⊗ Array2:
[[ 1  2  2  4  3  6]
 [ 3  4  6  8  9 12]
 [ 5  6 10 12 15 18]
 [ 4  8  5 10  6 12]


回到顶部