jax.experimental.sparse.CSR

class jax.experimental.sparse.CSR(args, *, shape)

JAX 中實作的實驗性 CSR 矩陣。

請注意：此類別與 JAX 轉換（如 grad 和 autodiff）的相容性極低，且功能非常有限。一般而言，您應該偏好 jax.experimental.sparse.BCOO。

此外，已知在 nse 大於所表示矩陣中真實的非零元素數量的情況下會發生錯誤。BCOO 能夠更好地處理這種情況。

參數:

shape (tuple[int, int])

__init__(args, *, shape)

方法

__init__(args, *, shape)
block_until_ready()
fromdense(mat, *[, nse, index_dtype])
sum(*args, **kwargs)
todense()
transpose([axes])
tree_flatten()
tree_unflatten(aux_data, children)

屬性

T
dtype
ndim
nse
size
data
indices
indptr
shape