jax.lax.full_like#

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

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

参数:

  • x (类数组对象 | DuckTypedArray) – 类似数组的示例，用于形状和 dtype 信息。

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

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

  • shape (Shape | None) – 可选，输出 ndarray 的 shape 参数。

  • sharding (Sharding | None) – 结果数组的可选分片规范。如果未指定，则输出将与输入具有相同的分片，但有一些例外/限制，特别是：1. 分片在跟踪期间不可用，因此这将依赖于 jit。2. 如果 x 是弱类型或未提交的，将使用默认分片。3. Shape 不是 None 并且与 x.shape 不同，将使用默认值。

返回:

一个与 x 具有相同形状的 ndarray，其条目设置为等于 fill_value，类似于 np.full 的输出。

返回类型:

Array