jax.experimental.sparse.CSC

jax.experimental.sparse.CSC#

class jax.experimental.sparse.CSC(args, *, shape)[source]#

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

参数::

shape (tuple[int, int])

__init__(args, *, shape)[source]#

方法

__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