- jax.experimental.sparse.bcoo_concatenate(operands, *, dimension)#
Sparse implementation of
BCOO]) – Sequence of BCOO arrays to concatenate. The arrays must have equal shapes, except in the dimension axis. Additionally, the arrays must have have equivalent batch, sparse, and dense dimensions.
int) – Positive integer specifying the dimension along which to concatenate the arrays. The dimension must be among batch or sparse dimensions of the input; concatenation along dense dimensions is not supported.
- Return type
A BCOO array containing the concatenation of the inputs.