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的返回。
(返回一个tuple,第一个元素是最大的value,第二个元素是对应的index)