mrpro.utils

Functions for tensor shaping, unit conversion, typing, etc.

Functions

broadcast_right(*x)	Broadcasting on the right.
fill_range_(tensor, dim)	Fill tensor in-place along a specified dimension with increasing integers.
ravel_multi_index(multi_index, dims)	Convert a multi-dimensional index into a flat index.
reduce_view(x[, dim])	Reduce expanded dimensions in a view to singletons.
remove_repeat(tensor, tol)	Replace dimensions with all equal values with singletons.
reshape_broadcasted(tensor, *shape)	Reshape a tensor while preserving broadcasted (stride 0) dimensions where possible.
smap(function, tensor[, passed_dimensions])	Apply a function to a tensor serially along multiple dimensions.
split_idx(idx, np_per_block[, np_overlap, ...])	Split a tensor of indices into different blocks.
unsqueeze_left(x, n)	Unsqueeze multiple times in the leftmost dimension.
unsqueeze_right(x, n)	Unsqueeze multiple times in the rightmost dimension.
unsqueeze_tensors_left(*x[, ndim])	Unsqueeze tensors on the left to the same number of dimensions.
unsqueeze_tensors_right(*x[, ndim])	Unsqueeze tensors on the right to the same number of dimensions.
zero_pad_or_crop(data, new_shape[, dim])	Change shape of data by center cropping or symmetric zero-padding.

Classes

TensorAttributeMixin

Create tensor attributes as buffer.

Modules

slice_profiles	Slice Profiles.
typing	Some type hints that are used in multiple places in the codebase but not part of mrpro's public API.
unit_conversion	Conversion between different units.