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