bartorch.linop.concatenate

Contents

bartorch.linop.concatenate#

bartorch.linop.concatenate(ops, axis=0)#

Apply every operator to the same input and lay the results end to end.

torch.cat of what they return, and what pylops calls VStack: the operators share a domain, and their codomains agree except along axis, where the sizes add up.

Parameters:

  • ops (sequence of LinearOperator) – At least one. All must have the same domain.

  • axis (int) – Which axis of the codomain grows, as an index into it.

Examples

>>> A = concatenate([FFT(shape, axes=-1), Identity(shape)])
>>> A.oshape == (2 * shape[0],) + shape[1:]
True
Contents