Hi,
for RL purposes I want to calculate the loss from MSE from the inputs x and x_expected: loss(x, x_expected), instead of loss(y_pred, y_expected). In the latter case the gradients are changing over time, but when using the inputs, it stalls. Anything I have to consider? Like copying grad_fn from the output / y_pred onto x so that the loss recognizes the graph and so on?