H_{in}, W_{in})
  * Output: (N, C_{out}, D_{out}, H_{out}, W_{out}) or (C_{out},
    D_{out}, H_{out}, W_{out}), where

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

     H_{out} = (H_{in} - 1) \times \text{stride}[1] - 2 \times
     \text{padding}[1] + \text{dilation}[1]           \times
     (\text{kernel\_size}[1] - 1) + \text{output\_padding}[1] + 1

     W_{out} = (W_{in} - 1) \times \text{stride}[2] - 2 \times
     \text{padding}[2] + \text{dilation}[2]           \times
     (\text{kernel\_size}[2] - 1) + \text{output\_padding}[2] + 1

Variables:
      * weight (Tensor) -- the learnable weights of the module
        of shape (\text{in_channels},
        \frac{\text{out_channels}}{\text{groups}},
        \text{kernel_size[0]}, \text{kernel_size[1]},