Handling ConvTranspose2d layers properly

[eldarkurtic]

Weights in the Conv2d layer are stored as tensors with shape (out_channels, in_channels, kernel_size[0], kernel_size[1]), while weights in the ConvTranspose2d are stored as tensors with shape (in_channels, out_channels, kernel_size[0], kernel_size[1]).