jax.lax.full

目录

jax.lax.full#

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

返回一个用 fill_value 填充的 shape 数组。

参数:

  • shape (Shape) – 整数序列，描述输出数组的形状。

  • fill_value (ArrayLike) – 用于填充新数组的值。

  • dtype (DTypeLike | None) – 输出数组的类型，或 None。如果不是 None，fill_value 将被转换为 dtype。

  • sharding (Sharding | None) – 结果数组的可选分片规范，注意，在即时编译模式下，分片当前将被忽略，这在未来可能会改变。

返回类型:

Array

目录