diff --git a/scripts/tcd/ANN_Data_Generator.py b/scripts/tcd/ANN_Data_Generator.py
index 33f7cf3ab1d7d70d2de2d9c8156b2a84f1c8ea57..184a93dad910942100c5607f1c7b26b456d273f9 100644
--- a/scripts/tcd/ANN_Data_Generator.py
+++ b/scripts/tcd/ANN_Data_Generator.py
@@ -260,11 +260,9 @@ class TrainingDataGenerator:
             Array containing normalized input data.
 
         """
-        normalized_input_data = []
-        for entry in input_data:
-            max_function_value = max(max(np.absolute(entry)), 1)
-            normalized_input_data.append(entry / max_function_value)
-        return np.array(normalized_input_data)
+        normalized_input_data = input_data/np.maximum(
+            1, np.max(np.abs(input_data), axis=1))[:, np.newaxis]
+        return normalized_input_data
 
     @staticmethod
     def _save_data(directory, data):