bartorch.linop.block_diag

Contents

bartorch.linop.block_diag#

bartorch.linop.block_diag(ops, axis=0, domain_axis=None)#

Give each operator its own part of the input and its own part of the output.

BART’s linop_stack, folded over the sequence. The domains are laid end to end along domain_axis and the codomains along axis; both default to the first, and each side must agree off the axis it grows along.

Parameters:

  • ops (sequence of LinearOperator) – At least one.

  • axis (int) – Which axis of the codomain grows.

  • domain_axis (int, optional) – Which axis of the domain grows; axis when left out.

Contents