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
