Common functions of pytorch

1.torch.max(input, dim, keepdim=False, *, out=None)

Returns a named tuple where is the given dimensionMedium Tensor The maximum value of each row.and is the index position (argmax) of each maximum found.(values,indices)valuesinputdimindices

If keepdim is True, the output tensor is the same as the inputThe tensors are of the same size, but the output tensor in dimension dim is of size 1.Otherwise, dim is compressed (see torch.squege()), resulting in the output tensor having one less dimension than the input tensor.