NestedIters

Create nditers for use in nested loops.

Abstract Signature:

NestedIters(op, axes)

PyTorch

API: —
Strategy: Custom / Partial

JAX (Core)

API: jax.numpy.nested_iters
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.nested_iters
Strategy: Direct Mapping
Official Docs ↗

Keras

API: —
Strategy: Custom / Partial

TensorFlow

API: —
Strategy: Custom / Partial

Apple MLX

API: —
Strategy: Custom / Partial