jax.lax.broadcasted_iota# jax.lax.broadcasted_iota(dtype, shape, dimension)[源代码][源代码]# 围绕 iota 的便捷包装器。 参数: dtype (DTypeLike) shape (Shape) dimension (int) 返回类型: Array