Task: PyTorch torch.max over Multiple Dimensions

Task: PyTorch torch.max over Multiple Dimensions: a task in Terminal-Lego-15k (Harbor dataset). Write a Python script that computes the maximum value of a PyTorch tensor over multiple dimensions simultaneously.