pymc.math.block_diagonal

pymc.math.block_diagonal(matrices, sparse=False, format='csr')

See pt.slinalg.block_diag or pytensor.sparse.basic.block_diag for reference

Parameters:

matrices: tensors

format: str (default ‘csr’)

must be one of: ‘csr’, ‘csc’

sparse: bool (default False)

if True return sparse format

Returns:

matrix