Calculates the output dimensions of a PyTorch conv2d, conv2d.T, conv3d or conv3d.T, layer.
adillwma / pytorch-conv-net-output-size-calculator Goto Github PK
View Code? Open in Web Editor NEWCalculates the output dimensions of a PyTorch conv2d, conv2d.T, conv3d or conv3d.T, layer.
License: MIT License