jax.lax.full_like

目录

jax.lax.full_like#

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

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

参数:

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

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

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

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

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

返回:

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

返回类型:

Array

目录