Hi,
I created an autograd functions, and in the forward / backward, I did numpy operations, and the function is returning cpu tensors.
This is not compatible with the rest of the code, which is using cuda tensors.
I wondered what was best :  Convert all my operations to cuda tensors inside the autograd function, or convert the cpu tensor to a cuda tensor after it was returned by my autograd function ?