jax.lax.full

jax.lax.full(shape, fill_value, dtype=None, *, sharding=None)

返回一个用 fill_value 填充的 shape 数组。

参数:

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

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

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

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

返回类型:

数组