torch.max will return the values and indices for the specified dimension, so you might want to access the desired attribute.
Besides that you would get the result containing the max. pixel values from all crops.