4D Tensor indexing with `gather()`
Currently I would like to perform this numpy equivalent command <SCODE>a = np.random.randn(5, 4, 3, 3)
a[range(5), :, [0, 1, 1, 2, 0], [1, 2, 0, 1, 0]]
<ECODE>
zuoxingdong
fmassa