jax.lax.full_like

内容

jax.lax.full_like#

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

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

参数：

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

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

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

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

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

返回:

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

返回类型:

数组

内容