scipy.sparse.kron

scipy.sparse.kron(A, B, format=None)

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

format of the result (e.g. “csr”)

Returns

kronecker product in a sparse matrix format

Examples

>>> from scipy import sparse
>>> A = sparse.csr_matrix(np.array([[0, 2], [5, 0]]))
>>> B = sparse.csr_matrix(np.array([[1, 2], [3, 4]]))
>>> sparse.kron(A, B).toarray()
array([[ 0,  0,  2,  4],
       [ 0,  0,  6,  8],
       [ 5, 10,  0,  0],
       [15, 20,  0,  0]])

>>> 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]])