Well I guess I dont know what I am doing here. I am confused.
I wanted to define my function using torch library, because I want the output to be tensorFloat, now I need to do backprob, so what I should do? how should I define this function?
Can you help me with defining this function the way that autograd works for it?