TorchLean API

NN.Tensor.Internal.Elab.Einsum.Kernel.Analysis

Einsum contraction-invariance analysis #

This module recognizes operands whose physical reads are constant across all contracted coordinates, including singleton-broadcast axes.

Report whether an operand is constant across every contracted coordinate.

Output axes never vary inside the contraction. A contracted physical axis is also constant when its dimension reduces to one, because broadcasting always reads its sole entry.

Instances For