jax.experimental.sparse.bcoo_concatenate#

jax.experimental.sparse.bcoo_concatenate(operands,*,dimension)[source]#

Sparse implementation ofjax.lax.concatenate()

Parameters:

• operands (Sequence[BCOO]) – Sequence of BCOO arrays to concatenate. The arrays must have equalshapes, except in thedimension axis. Additionally, the arrays must havehave equivalent batch, sparse, and dense dimensions.

• dimension (int) – Positive integer specifying the dimension along which to concatenatethe arrays. The dimension must be among batch or sparse dimensions of the input;concatenation along dense dimensions is not supported.

Returns:

A BCOO array containing the concatenation of the inputs.

Return type:

BCOO