jax.lax.full_like#

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

创建一个类似于 np.full 的全数组，基于示例数组 x。

参数:

  • x (ArrayLike | DuckTypedArray) – 示例类数组，用于形状和数据类型信息。

  • fill_value (ArrayLike) – 用于填充输出数组条目的标量值。

  • dtype (DTypeLike | None | None) – 可选，输出 ndarray 的数据类型参数。

  • 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