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