tensorflow equivalent of torch.gather. I have a tensor of shape (16, 4096, 3) . I have another tensor of indices of shape (16, 32768, 3) .
確定! 回上一頁