TD3 (#338)

rl_coach/architectures/tensorflow_components/utils.py

@@ -38,3 +38,10 @@ def get_activation_function(activation_function_string: str):
         "Activation function must be one of the following {}. instead it was: {}" \
             .format(activation_functions.keys(), activation_function_string)
     return activation_functions[activation_function_string]
+
+
+def squeeze_tensor(tensor):
+    if tensor.shape[0] == 1:
+        return tensor[0]
+    else:
+        return tensor