jax.lax.iota# jax.lax.iota(dtype, size)[source]# 包装 XLA 的 Iota 运算符。 参数: dtype (DTypeLike) size (int) 返回值类型: 数组
