jax.experimental.sparse.BCSR

class jax.experimental.sparse.BCSR(args, *, shape, indices_sorted=False, unique_indices=False)

在 JAX 中实现的实验性批处理 CSR 矩阵。

参数:

• args (tuple[Array, Array, Array])

• shape (Shape)

• indices_sorted (bool)

• unique_indices (bool)

__init__(args, *, shape, indices_sorted=False, unique_indices=False)

参数:

• args (tuple[Array, Array, Array])

• shape (Sequence[int])

• indices_sorted (bool)

• unique_indices (bool)

方法

__init__(args, *, shape[, indices_sorted, ...])
block_until_ready()
from_bcoo(arr)
from_scipy_sparse(mat, *[, index_dtype, ...])	从 scipy.sparse 数组创建一个 BCSR 数组。
fromdense(mat, *[, nse, index_dtype, ...])	从（密集的）:class:数组 创建一个BCSR数组。
sum(*args, **kwargs)
sum_duplicates([nse, remove_zeros])	返回一个数组副本，其中重复的索引值被求和。
to_bcoo()
todense()	创建数组的密集版本。
transpose(*args, **kwargs)
tree_flatten()
tree_unflatten(aux_data, children)

属性

T
dtype
n_batch
n_dense
n_sparse
ndim
nse
size
data
indices
indptr
shape
indices_sorted
unique_indices