jax.lax.iota#

jax.lax.iota(dtype, size)[原始碼]#

包裝 XLA 的 Iota 運算子。

參數:

  • dtype (DTypeLike)

  • size (int)

回傳型別:

Array