Corrected bug on CustomHingeLoss where it would not run on gpu

torch_modules/src/CustomHingeLoss.cpp

 #include "CustomHingeLoss.hpp"
+#include "NeuralNetwork.hpp"
 
 torch::Tensor CustomHingeLoss::operator()(torch::Tensor prediction, torch::Tensor gold)
 {
-  torch::Tensor loss = torch::zeros(1);
+  torch::Tensor loss = torch::zeros(1).to(NeuralNetworkImpl::device);
 
   for (unsigned int i = 0; i < prediction.size(0); i++)
   {