1\right\rfloor
       H_{out} = \left\lfloor\frac{H_{in} + 2 \times
       \text{padding}[1] - \text{dilation}[1] \times
       (\text{kernel\_size}[1] - 1) - 1}{\text{stride}[1]} +
       1\right\rfloor

       W_{out} = \left\lfloor\frac{W_{in} + 2 \times
       \text{padding}[2] - \text{dilation}[2] \times
       (\text{kernel\_size}[2] - 1) - 1}{\text{stride}[2]} +
       1\right\rfloor

Examples:
  >>> # pool of square window of size=3, stride=2
  >>> m = nn.MaxPool3d(3, stride=2)
  >>> # pool of non-square window
  >>> m = nn.MaxPool3d((3, 2, 2), stride=(2, 1, 2))
  >>> input = torch.randn(20, 16, 50, 44, 31)
  >>> output = m(input)
