Update input_preprocess.py

input_preprocess.py

@@ -371,4 +371,4 @@ def create_labels(task, scenario_names, n_beams=64):
     for scenario_name in scenario_names:
         data = DeepMIMO_data_gen(scenario_name)
         labels.extend(label_gen(task, data, scenario_name, n_beams=n_beams))
-    return labels
+    return torch.tensor(labels)