Tensor.gather ) is a multi-index selection method. Look at the following example from the official docs: t = torch.tensor([[1,2],[3,4]]) r = torch.gather(t, ...
確定! 回上一頁