Sync public subset from Flux

2025-10-20 12:24:21 +00:00
parent 9a69d64d79
commit a334b74935
8 changed files with 230 additions and 35 deletions
--- a/include/modules/neural_networks/loss/Loss.h
+++ b/include/modules/neural_networks/loss/Loss.h
@@ -13,9 +13,11 @@ namespace neural_networks{
 	struct Loss{

 			utils::Vector<Td> sample_losses;
+			utils::Matrix<Td> dinputs;
 			Td data_loss;

 			virtual utils::Vector<Td> forward(const utils::Matrix<Td>& output, const utils::Matrix<Ti>& y) = 0;
+			virtual void backward(const utils::Matrix<Td>& dvalues, const utils::Matrix<Ti>& y) = 0;

 			Td calculate(const utils::Matrix<Td>& output, const utils::Matrix<Ti>& y){