结论:
torch.Tensor.argmax(dim:int,keepdim:bool)->Tensor
找Tensor中的指定维度dim上最大值所在下标,返回是一个Tensor。keepdim指定返回的Tensor是否和原来的一样形状。若为False,则在dim维度上被压缩。
torch.Tensor.max(dim:int)->Tuple
找Tensor中的指定维度dim上最大值所在下标,返回一个Tuple,第一个元素是值Tensor,指定最大值。第二个元素则为类似argmax的返回。