tensordot

sparse.tensordot(a, b, axes=2)[source]

Perform the equivalent of numpy.tensordot.

Parameters:
  • b (a,) – The arrays to perform the tensordot operation on.
  • axes (tuple[Union[int, tuple[int], Union[int, tuple[int]], optional) – The axes to match when performing the sum.
Returns:

The result of the operation.

Return type:

Union[COO, numpy.ndarray]

See also

numpy.tensordot()
NumPy equivalent function