jax.lax.full

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

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

参数:

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

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

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

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

返回类型:

数组