Hello, I have a problem with torch.max operation.

When I used torch.max with 1x1x1xn tensor, like

```
import torch
a = torch.randn(1, 1, 1, 5)
a.max(0)
```

It produces a tuple of two 1x1x1xn tensor. (One for maximum value, and another for indices)

However, if i try the same operation with n = 1, like

```
import torch
b = torch.randn(1, 1, 1, 1)
b.max(0)
```

torch.max produces a tuple of two **1x1x1** tensor.

On the other hand

```
import torch
c = torch.randn(2, 1, 1, 1)
c.max(0)
```

it produces a tuple of two **1x1x1** tensor.

I think **b** and **c** work correctly, and

```
a.max(0)
```

should produce a tuple of **1x1xn** tensor. What’s wrong with it?