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 (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

Try it in your browser!

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

Go BackOpen In Tab