hi everyone, I have been reading posts here for awhile, but this is my first topic in this friendly community ^^
I’m interested in pytorch’s convolution layer:
mainly in estimating the number of floating point operations, time and additional space complexity…
-so far my floating point operations calculation was derived strictly from the mathematical equation, and I got the following expression: FP_Ops=HWC_out*(2C_inK^2 + 1).
-I didn’t find the actual conv2d implementation code under the “nn/functional.py” file,
I was hoping for your help in finding the time and additional space complexity (or a direction to the source code is possible).
Thanks in advance and have a great week!