I’m sorry for digging out this topic but I didn’t want to create a new one for a simple question.
Why is grad_output a tuple which length is the batch size but grad_input contains the batch as its first dimension ? What’s behind this design choice ?
Thanks !