I want to get the tensor of the same shape as calculated by pytorch. In other words, I want to know the tensorflow equivalent of torch.gather.
確定! 回上一頁