mm.math.tensordot

math.tensordot(b, axes)

The tensordot product of a and b.

Parameters:

• a (ndarray[Tuple[int, ...], Union[TypeVar(R, float16, float32, float64), TypeVar(C, complex64, complex128), TypeVar(Z, int16, int32, int64), TypeVar(N, uint16, uint32, uint64)]]) – The first array to take the tensordot product of

• b (ndarray[Tuple[int, ...], Union[TypeVar(R, float16, float32, float64), TypeVar(C, complex64, complex128), TypeVar(Z, int16, int32, int64), TypeVar(N, uint16, uint32, uint64)]]) – The second array to take the tensordot product of

• axes (Sequence[int]) – The axes to take the tensordot product over

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 tensordot product of a and b

code/api/mrmustard.math.tensordot

 

Download Python script

 

Download Notebook

 

View on GitHub