jax.lax.full_like#

jax.lax.full_like(x, fill_value, dtype=None, shape=None, sharding=None)[來源]#

根據範例陣列 x 建立類似 np.full 的完整陣列。

參數:

  • x (ArrayLike | DuckTypedArray) – 範例類陣列，用於形狀和 dtype 資訊。

  • fill_value (ArrayLike) – 純量值，用於填滿輸出陣列的條目。

  • dtype (DTypeLike | None | None) – 選用，輸出 ndarray 的 dtype 參數。

  • shape (Shape | None | None) – 選用，輸出 ndarray 的形狀參數。

  • sharding (Sharding | None | None) – 結果陣列的選用分片規格。若未指定，輸出將與輸入具有相同的分片，但有一些例外/限制，特別是：1. 分片在追蹤期間不可用，因此這將依賴 jit。2. 如果 x 是弱類型或未提交，將使用預設分片。3. Shape 不是 None 且與 x.shape 不同，將使用預設值。

傳回:

與 x 具有相同形狀的 ndarray，其條目設定為等於 fill_value，類似於 np.full 的輸出。

傳回類型:

Array