jax.experimental.sparse.todense

Contents

jax.experimental.sparse.todense#

jax.experimental.sparse.todense(arr)[source]#

Convert input to a dense matrix. If input is already dense, pass through.

Parameters:

arr (JAXSparse | Array)

Return type:

Array