mm.math.einsum

math.einsum(*tensors)

The result of the Einstein summation convention on the tensors.

Parameters:

• string (str) – The string of the Einstein summation convention.

• tensors – The tensors to perform the Einstein summation on.

Return type:

ndarray[Tuple[int, ...], Union[TypeVar(R, float16, float32, float64), TypeVar(C, complex64, complex128), TypeVar(Z, int16, int32, int64), TypeVar(N, uint16, uint32, uint64)]]

Returns:

The result of the Einstein summation convention.

code/api/mrmustard.math.einsum

 

Download Python script

 

Download Notebook

 

View on GitHub