Tf.gather functionallity missing in pytorch

[nima_rafiee]

Yes. torch.gather() is different. but I think you just can use direct indexing of pytorch.