Thank you for your suggestions! Actually what I am trying to do is to replace the multiplications and additions in the convolutions with my custom operations (like myadd, mymult) for some purposes so I am trying to find a way to do so… (I only need to replace during forward pass though)