tensordot¶

sparse.tensordot(a, b, axes=2, *, return_type=None)[source]¶

Perform the equivalent of numpy.tensordot.

Parameters

a (Union[COO, np.ndarray, scipy.sparse.spmatrix]) – The arrays to perform the tensordot operation on.
b (Union[COO, np.ndarray, scipy.sparse.spmatrix]) – 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.
return_type ({None, COO, np.ndarray}, optional) – Type of returned array.

Returns

The result of the operation.

Return type

Union[COO, numpy.ndarray]

Raises

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