jax.lax.iota

jax.lax.iota(dtype, size)

封装 XLA 的 Iota 运算符。

参数:

• dtype (DTypeLike)

• size (int)

返回类型:

Array