Thanks to your reply.
Sorry. my question was not clear.
I want to make custom CUDA layer in my model
It means to make (.cu) file.
So, I have to use Aten tensor which means I can’t use autograd.
I have to make custom forward and backward layer for getting right gradient.
I found some example for make custom forward backwadr layer.
But, I can’t find how to update variables in c++
Can I have to make custom optimizer to update variables or Is there simple way.