mlx.core.einsum

Contents

mlx.core.einsum#

einsum(subscripts: str, *operands, stream: StreamOrDevice = None) → array#

Perform the Einstein summation convention on the operands.

Contractions are implemented with matmul() and have the same requirements.

Parameters:

  • subscripts (str) – The Einstein summation convention equation.

  • *operands (array) – The input arrays.

Returns:

The output array.

Return type:

array

Contents