ArrayIterator¶

Iterator over an array’s first dimension.

Abstract Signature:

ArrayIterator(x: Tensor)

PyTorch

API: iter
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.ArrayIterator
Strategy: Direct Mapping
Official Docs ↗