jax.experimental.sparse.CSC#

class jax.experimental.sparse.CSC(args, *, shape)[源代码]#

JAX 中实现的实验性 CSC 矩阵；API 可能发生变化。

参数:

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