jax.lax.broadcasted_iota# jax.lax.broadcasted_iota(dtype, shape, dimension, sharding=None)[原始碼]# iota 的便利包裝函式。 參數: dtype (DTypeLike) shape (Shape) dimension (int) 返回類型: Array
