jax.lax.iota# jax.lax.iota(dtype, size)[源代码][源代码]# 封装了 XLA 的 Iota 操作符。 参数: dtype (DTypeLike) size (int) 返回类型: Array
