"kernel_size"
  * **padding** (*Union**[**int**, **Tuple**[**int**, **int**,
    **int**]**]*) -- Implicit negative infinity padding to be
    added on all three sides

  * **dilation** (*Union**[**int**, **Tuple**[**int**, **int**,
    **int**]**]*) -- a parameter that controls the stride of
    elements in the window

  * **return_indices** (*bool*) -- if "True", will return the max
    indices along with the outputs. Useful for
    "torch.nn.MaxUnpool3d" later

  * **ceil_mode** (*bool*) -- when True, will use *ceil* instead
    of *floor* to compute the output shape

Shape:
      * Input: (N, C, D_{in}, H_{in}, W_{in}) or (C, D_{in}, H_{in},
        W_{in}).
  * Output: (N, C, D_{out}, H_{out}, W_{out}) or (C, D_{out},
    H_{out}, W_{out}), where

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