cat

iskra.sparse.cat(xs: Sequence[SparseTensor], dim: int = 0) → SparseTensor[SOURCE]

Concatenates sparse COO matrices along a dimension.

Parameters:

• xs (Sequence[SparseTensor]) – Sequences of matrices to cocnatenate. Matrix shapes must match in all dimensions except dim.

• dim (int, optional) – Dimension to concatenate in. Defaults to 0.

Raises:

ValueError – Throws an exception if shapes do not match in all dimensions except dim.

Returns:

​ (SparseTensor) – Concatenated sparse COO matrix.