CNN Dimensions Calculator
Quickly calculate the output shape from a CNN (using nn.Conv2d/ConvTranspose2d/*Pool2d layers), given the shape of input tensor.
0 comments
write a comment
You need to register and be logged in to comment to this post.