jax.experimental.sparse.JAXSparse

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

高级 JAX 稀疏对象的基类。

参数:

• args (tuple[Array, ...])

• shape (tuple[int, ...])

__init__(args, *, shape)

参数:

• args (tuple[Array, ...])

• shape (Sequence[int])

方法

__init__(args, *, shape)
block_until_ready()
sum(*args, **kwargs)
transpose([axes])
tree_flatten()
tree_unflatten(aux_data, children)

属性

T
ndim
size
data
shape
nse
dtype