jax.lax.iota#

jax.lax.iota(dtype, size)[源代码]#

包装 XLA 的 Iota 运算符。

参数:

  • dtype (DTypeLike)

  • size (int)

返回类型:

Array