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, ...])	从 (密集) Array 创建 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