Added generic pytorch implementation for multi layer linear NN.

custom_types.py

@@ -0,0 +1,22 @@
+import torch
+from typing import Tuple, NamedTuple, List, Callable
+import numpy as np
+
+TrainingBatch = Tuple[List[List[float]], List[int]]
+
+
+class LossFun(NamedTuple):
+    exec: Callable[[np.array, np.array], float]
+    deriv: Callable[[np.array, np.array], np.array]
+
+
+class EvaluationResults(NamedTuple):
+    total: int
+    correct: int
+    accumulated_loss: float
+
+
+if torch.cuda.is_available():
+    DEVICE = "cuda:0"
+else:
+    DEVICE = "cpu"