Utility function for calculating the shape of a conv output

vision
3

I imagine if you wanted a function you could generate a random input, pass it to conv2d, and check the size of the output (this is definitely not efficient but works as a sanity check)

2 Likes