jax.experimental.sparse.empty

目录

jax.experimental.sparse.empty#

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

创建一个空的稀疏数组。

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

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

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

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

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

  • sparse_format (str)

返回:

空稀疏矩阵。

返回类型:

mat