Calculates output shape of convolutional layers according to torch.nn.Conv2d of Pytorch.
liguiye / conv2dshapecalculator Goto Github PK
View Code? Open in Web Editor NEWPytorch torch.nn.Conv2d Shape Calculator
Home Page: https://liguiye.github.io/Conv2dShapeCalculator/