Corrected bug where tensor was not initialized to the correct device

torch_modules/src/LossFunction.cpp

 #include "LossFunction.hpp"
 #include "util.hpp"
+#include "NeuralNetwork.hpp"
 
 void LossFunction::init(std::string name)
 {
@@ -50,13 +51,13 @@ torch::Tensor LossFunction::getGoldFromClassesIndexes(int nbClasses, const std::
 
   if (index == 0 or index == 2 or index == 4)
   {
-    auto gold = torch::zeros(1, torch::TensorOptions(torch::kLong));
+    auto gold = torch::zeros(1, torch::TensorOptions(torch::kLong).device(NeuralNetworkImpl::device));
     gold[0] = goldIndexes.at(0);
     return gold;
   }
   if (index == 1 or index == 3)
   {
-    auto gold = torch::zeros(nbClasses, torch::TensorOptions(torch::kLong));
+    auto gold = torch::zeros(nbClasses, torch::TensorOptions(torch::kLong).device(NeuralNetworkImpl::device));
     for (auto goldIndex : goldIndexes)
       gold[goldIndex] = 1;
     return gold;