tensorflow 2, using tensor as index without for loop

Question

Let\'s say I have a tensor of shape [3000,20,10], I\'ll call it "input"... I have another tensor of shape [3000] that contains only indices for the 3rd dimension o