xarray_einstats.einops._rearrange(da, out_dims, in_dims=None, dim_lengths=None)[source]#

Wrap einops.rearrange.

This is the function that actually interfaces with einops. xarray_einstats.einops.rearrange is the user facing version as it exposes two possible APIs, one of them significantly less verbose and more friendly (but much less flexible).


Input DataArray to be rearranged

out_dimslist of str, list or dict

See docstring of rearrange

in_dimslist of str or dict, optional

See docstring of rearrange

dim_lengthsdict, optional

kwargs with key equal to dimension names in out_dims (that is, strings or dict keys) are passed to einops.rearrange the rest of keys are passed to xarray.apply_ufunc