mrpro.utils.ravel_multi_index

mrpro.utils.ravel_multi_index(multi_index: Sequence[Tensor], dims: Sequence[int]) → Tensor[source]

Convert a multi-dimensional index into a flat index.

Parameters:

• multi_index (Sequence[Tensor]) – Sequence of integer index tensors.

• dims (Sequence[int]) – The shape of the tensor being indexed.

Returns:

index – Flattened index.