What is the difference between nn.functional.conv_transpose2d and nn.ConvTranspose2d?

1

Hi,
What is the difference between nn.functional.conv_transpose2d and nn.ConvTranspose2d?

Is nn.ConvTranspose2d used for training and nn.functional.conv_transpose2d just used for visualization?

2

There is no difference.
One is the module interface (nn.ConvTranspose2d), while the other is the functional interface.
Also, note that nn.ConvTranspose2d uses functional.conv_transpose2d internally, and the module interface is simply a convenience interface that creates the weights and biases for you

3 Likes