einsum

sparse.einsum(*operands, **kwargs)[source]

Perform the equivalent of numpy.einsum.

Parameters:
  • subscripts (str) – Specifies the subscripts for summation as comma separated list of subscript labels. An implicit (classical Einstein summation) calculation is performed unless the explicit indicator ‘->’ is included as well as subscript labels of the precise output form.

  • operands (sequence of SparseArray) – These are the arrays for the operation.

  • dtype (data-type, optional) – If provided, forces the calculation to use the data type specified. Default is None.

  • **kwargs (dict, optional) – Any additional arguments to pass to the function.

Returns:

output – The calculation based on the Einstein summation convention.

Return type:

SparseArray