I am a begineer in pytorch. I have a question about “grad_output” is how to compute by mse loss in pytorch.In my case, I define a new layer as network output layer , and also my own forward and backward function but don’t have network parameters, i just want to do use “grad_output” to do some work . but i don’t known grad_output is how to compute by mse loss . Can you give me some advice about this question ?
Thank you !
class MyFun(torch.autograd.Function):
def forward(self, inp):
return inp
def backward(self, grad_output):
"
use grad_output
""
return grad
loss = torch.sum((y_pred - y).pow(2))
loss.backward()
type or paste code here