torch.argmax Inconsistent with torch.max() Returns index of last(02) Save Image Image gallery: Torch.maximum