Fix dimension mismatch in MulticoreBPFLayer

scripts/deep_learning_model.py

@@ -160,7 +160,7 @@ def call(self, inputs):
             raise ValueError(f"Dimension mismatch: reshaped inputs shape {input_shape[1:]} does not match predicted_measurements shape {predicted_shape}")
 
         # Reshape inputs to match the shape of predicted_measurements
-        reshaped_inputs = tf.reshape(inputs, [input_shape[0], predicted_shape[1], predicted_shape[0]])
+        reshaped_inputs = tf.reshape(inputs, [input_shape[0], predicted_shape[0], predicted_shape[1]])
         tf.print("Shape of reshaped_inputs:", tf.shape(reshaped_inputs))
 
         self.particle_weights.assign(tf.reduce_sum(tf.square(reshaped_inputs - tf.transpose(predicted_measurements)), axis=-1))