[water] index expression inference for attention

[ftynse]

Index expression inference should match the results of pywave's index sequence analysis for vanilla attention

[ftynse]

Done ending in #1171.