fix(autograd): gather/indexSelect backward and indexSelect forward reject rank>=2 indices - #995

Merged

michalharakal merged 2 commits into

fix/gather-indexselect-backward-multidim-indices

Aug 16, 2026

Commits on Aug 15, 2026

fix(autograd): gather/indexSelect backward and indexSelect forward reject rank>=2 indices
michalharakal
committed
8ac220b
test(autograd): edge cases and an end-to-end embedding training test for #994
michalharakal
committed
e4ba581