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 的输出。
- 返回类型: