jax.experimental.sparse.empty#

jax.experimental.sparse.empty(shape, dtype=None, index_dtype='int32', sparse_format='bcoo', **kwds)[source]#

Create an empty sparse array.

Parameters:
  • shape (Sequence[Union[int, Any]]) – sequence of integers giving the array shape.

  • dtype (Union[str, type[Any], dtype, SupportsDType, None]) – (optional) dtype of the array.

  • index_dtype (Union[str, type[Any], dtype, SupportsDType]) – (optional) dtype of the index arrays.

  • format – string specifying the matrix format (e.g. [‘bcoo’]).

  • **kwds – additional keywords passed to the format-specific _empty constructor.

  • sparse_format (str) –

Returns:

empty sparse matrix.

Return type:

mat