From 7c984c6fd2c9feee46251e66b32937d0579ab6e5 Mon Sep 17 00:00:00 2001
From: czzhangheng <cz189@qq.com>
Date: Mon, 24 Nov 2025 21:53:01 +0800
Subject: [PATCH] AEPSA v0.1

---
 model/AEPSA/aepsa.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/model/AEPSA/aepsa.py b/model/AEPSA/aepsa.py
index 409b585..0c13da4 100644
--- a/model/AEPSA/aepsa.py
+++ b/model/AEPSA/aepsa.py
@@ -227,11 +227,10 @@ class AEPSA(nn.Module):
         """
         x = x[..., :1]
         x_enc = rearrange(x, 'b t n c -> b n c t')
+        # 原版Patch
         enc_out, n_vars = self.patch_embedding(x_enc) # (B, N, C)
         # 应用图增强编码器（自动生成图结构）
         graph_enhanced = self.graph_encoder(enc_out)
-        # 保持相同的维度
-        
         # 特征融合 - 现在两个张量都是三维的 [B, N, d_model]
         enc_out = torch.cat([enc_out, graph_enhanced], dim=-1)
         enc_out = self.feature_fusion(enc_out)