Edit dimensions

2025-12-16 18:56:49 +01:00
parent 5895bbaed3
commit f4b28ee6fb
1 changed files with 7 additions and 1 deletions
@@ -65,8 +65,14 @@ with torch.no_grad():
            X_c = X_c.to(DEVICE)
        y_pred, _ = model(X_t, X_f, X_c, warm // step, pred // step)
        lat_pred, lon_pred, alt_pred = denorm_coords(y_pred[...,0], y_pred[...,1], y_pred[...,2])
+        lat_pred = lat_pred.flatten()
+        lon_pred = lon_pred.flatten()
+        alt_pred = alt_pred.flatten()
        lat_true, lon_true, alt_true = denorm_coords(y[...,0], y[...,1], y[...,2])
-        results.append(torch.stack([lat_true, lon_true, alt_true, lat_pred, lon_pred, alt_pred], dim=1).cpu())
+        lat_true = lat_true.flatten()
+        lon_true = lon_true.flatten()
+        alt_true = alt_true.flatten()
+        results.append(torch.cat([lat_true, lon_true, alt_true, lat_pred, lon_pred, alt_pred], dim=-1).cpu())
    
    np_y_all = torch.cat(results, dim=0).numpy()
    np.savetxt(