jax.experimental.sparse.bcoo_concatenate

目录

jax.experimental.sparse.bcoo_concatenate#

jax.experimental.sparse.bcoo_concatenate(operands, *, dimension)[源代码]#

jax.lax.concatenate() 的稀疏实现

参数:

operands (Sequence[BCOO]) – 要连接的 BCOO 数组序列。除了 dimension 轴外，数组必须具有相同的形状。此外，数组必须具有等效的批处理维度、稀疏维度和密集维度。
dimension (int) – 指定沿其连接数组的维度，正整数。该维度必须是输入的批处理维度或稀疏维度之一；不支持沿密集维度进行连接。

返回:

包含输入连接的 BCOO 数组。

返回类型: