jax.lax.full

内容

jax.lax.full#

jax.lax.full(shape, fill_value, dtype=None, *, sharding=None)[source]#

返回一个填充有 fill_value 的形状为 shape 的数组。

参数：

  • shape (Shape) – 整数序列，描述输出数组的形状。

  • fill_value (ArrayLike) – 用于填充新数组的值。

  • dtype (DTypeLike | None | None) – 输出数组的类型，或 None。如果非 None，则 fill_value 将被转换为 dtype。

  • sharding (Sharding | None | None) – 结果数组的可选分片规范，注意，在 jitted 模式下，分片将被忽略，这在将来可能会改变。

返回类型：

数组

内容