kron — SciPy v1.15.2 Manual (original) (raw)

scipy.sparse.

scipy.sparse.kron(A, B, format=None)[source]#

kronecker product of sparse matrices A and B

Parameters:

Asparse or dense matrix

first matrix of the product

Bsparse or dense matrix

second matrix of the product

formatstr, optional (default: ‘bsr’ or ‘coo’)

format of the result (e.g. “csr”) If None, choose ‘bsr’ for relatively dense array and ‘coo’ for others

Returns:

kronecker product in a sparse format.

Returns a sparse matrix unless either A or B is a

sparse array in which case returns a sparse array.

Examples

import numpy as np import scipy as sp A = sp.sparse.csr_array(np.array([[0, 2], [5, 0]])) B = sp.sparse.csr_array(np.array([[1, 2], [3, 4]])) sp.sparse.kron(A, B).toarray() array([[ 0, 0, 2, 4], [ 0, 0, 6, 8], [ 5, 10, 0, 0], [15, 20, 0, 0]])

sp.sparse.kron(A, [[1, 2], [3, 4]]).toarray() array([[ 0, 0, 2, 4], [ 0, 0, 6, 8], [ 5, 10, 0, 0], [15, 20, 0, 0]])