einsum

ivy.einsum(equation, *operands, f=None)[source]

Sums the product of the elements of the input operands along dimensions specified using a notation based on the Einstein summation convention.

Parameters
  • equation (str) – A str describing the contraction, in the same format as numpy.einsum.

  • operands (seq of arrays) – the inputs to contract (each one an ivy.Array), whose shapes should be consistent with equation.

  • f (ml_framework, optional) – Machine learning framework. Inferred from inputs if None.

Returns

The array with sums computed.


Supported Frameworks:

empty jax_logo empty tf_logo empty pytorch_logo empty mxnet_logo empty numpy_logo empty