I am afraid that some of the operations I am doing in the forward (maybe the zero padding and unpadding) aren’t taken into account in the backward or raise some problem, that I should be writing them differently or add something, nuances… Maybe if there is a way I can check that everything is added appropriately in the backward it could help…
Tnx