einops_reduce

ivy.einops_reduce(x: Union[ivy.Array, ivy.NativeArray], pattern: str, reduction: Union[str, Callable], **axes_lengths: Dict[str, int]) → Union[ivy.Array, ivy.NativeArray][source]

Perform einops reduce operation on input array x.

Parameters
  • x (array) – Input array to be reduced.

  • pattern (str) – Reduction pattern.

  • reduction (str or callable) – One of available reductions (‘min’, ‘max’, ‘sum’, ‘mean’, ‘prod’), or callable.

  • axes_lengths (keyword parameter args) – Any additional specifications for dimensions.

Returns

New array with einops.reduce having been applied.


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty