My goal is to predict an integer array of size (batch_size x 23) eg: [12, 20, 30, ...., 34] using an array of size (batch_size x 230). So the final step is loss value calcul
