I have a tensor, a of size bsz * 10000
a tensor, idx of size bsz * 30 which contains long values - index values lying between 0 and 10000
a tensor, b of size bsz * 30 float values
how can I add values in b to tensor a along the dimension 2 (10000) at indices specified by idx ?