Module advanced_indexing 

Advanced tensor indexing operations

This module implements sophisticated indexing patterns including:

• Fancy indexing (NumPy-style advanced indexing)
• Multi-dimensional gather/scatter
• Index broadcasting
• Masked indexing

Use Cases

• Embeddings lookup in transformers
• Attention mechanisms
• Sparse gradient updates
• Dynamic batching
• Index-based data shuffling

Enums

ScatterMode

Scatter modes for accumulation

Functions

advanced_gather

Advanced gather operation with multi-dimensional support

advanced_scatter

Advanced scatter operation with multi-dimensional support

fancy_index_mask

Fancy indexing with boolean masks