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]

Raises

ValueError – If all arguments don’t have zero fill-values.

See also

numpy.tensordot()

NumPy equivalent function