enp.unflatten

etils.enp.unflatten(

array: etils.enp.einops_utils._ArrayT,

batch_shape: tuple[int, ...],

pattern: str,

) → etils.enp.einops_utils._ArrayT