AndrewZhaoLuo commented on pull request #9336: URL: https://github.com/apache/tvm/pull/9336#issuecomment-950006190
I think in general, `I = the dimension that will change when we expect the input channels to change`. `O = the dimension that will change when we expect the output channels to change`. In this operator, this is not the case (and it also isn't for the 1D and 3D cases too probably but I will change those if this gets good traction). I will say I can understand maybe why this convention is so weird. conv2d_transpose can be seen as the gradient of conv2d. So 'I' and 'O' represent the input and output channels of the original conv2d. However, in conv2d_transpose, we don't have this context really and flipping the signs will cause confusion. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
