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。如果不是 None，fill_value 將會被轉換為 dtype。

  • sharding (Sharding | None | None) – 結果陣列的可選分片規格，請注意，目前在 jitted 模式下分片將被忽略，未來可能會更改。

傳回類型:

Array