Resolve symbolic shapes for neural network layers
Optional
Resolve attention layer shapes
Get the underlying environment
Resolve linear layer shapes: [batch, in_features] -> [batch, out_features]
Resolve symbolic shapes for neural network layers