tensorflow equivalent of torch.gather

来源：https://stackoverflow.com/questions/52129909/tensorflow-equivalent-of-torch-gather