TorchLean API

NN.Tensor.Internal.Elab.Einsum.Contraction.Index

Certified native-index normalization #

This module recognizes native coordinate arithmetic emitted by the einsum compiler and replaces conversion and quotient-remainder round trips with proved equal expressions.

Report whether a generated kernel contains a conversion or quotient-remainder expression handled by the focused native-index simplifier.

The scan keeps the simplifier away from unrelated symbolic and scalar terms.

Instances For

    Normalize generated native-index expressions in two certified phases.

    The first pass removes native-to-natural-to-native conversions. The second then sees complete quotient-remainder reconstructions and replaces them with direct native indexing. The returned equality composes both simplifier certificates.

    Instances For