jax.experimental.sparse.empty

内容

jax.experimental.sparse.empty#

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

创建一个空的稀疏数组。

参数:
  • shape (Shape) – 给定数组形状的整数序列。

  • dtype (DTypeLike | None | None) – (可选)数组的 dtype。

  • index_dtype (DTypeLike) – (可选)索引数组的 dtype。

  • format – 指定矩阵格式的字符串(例如 [‘bcoo’])。

  • **kwds – 传递给特定于格式的 _empty 构造函数的额外关键字。

  • sparse_format (str)

返回值:

空的稀疏矩阵。

返回类型:

mat