Update how loss function work
Context
Currently, Loss function only compute loss but do not create the gradient of the graph output node.
This MR aims at fixing this issue following the solution described in: #1 (closed)
Modified files
-
MSE
function now set themGrad
of the pred Tensor. - Bind loss function MSE
Edited by Cyril Moineau