From 369db12fc68059a0f69c84a135a75a0dc35159a3 Mon Sep 17 00:00:00 2001
From: yukun-hh <zhangyukunhh@gmail.com>
Date: Thu, 5 Mar 2026 15:31:25 +0800
Subject: [PATCH] py study

---
 main.py | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index eb389a0..6cb977a 100644
--- a/main.py
+++ b/main.py
@@ -2,15 +2,28 @@
 
 # 按 Shift+F10 执行或将其替换为您的代码。
 # 按 双击 Shift 在所有地方搜索类、文件、工具窗口、操作和设置。
+import torch
+from torch import nn
+# 按 Ctrl+F8 切换断点。
+class MyDictDense(nn.Module):
+    def __init__(self):
+        super(MyDictDense, self).__init__()
+        self.params = nn.ParameterDict({
+                'linear1': nn.Parameter(torch.randn(4, 4)),
+                'linear2': nn.Parameter(torch.randn(4, 1))
+        })
+        self.params.update({'linear3': nn.Parameter(torch.randn(4, 2))}) # 新增
 
+    def forward(self, x, choice='linear1'):
+        return torch.mm(x, self.params[choice])
 
-def print_hi(name):
-    # 在下面的代码行中使用断点来调试脚本。
-    print(f'Hi, {name}')  # 按 Ctrl+F8 切换断点。
 
 
 # 按装订区域中的绿色按钮以运行脚本。
 if __name__ == '__main__':
-    print_hi('PyCharm')
 
+
+
+    net = MyDictDense()
+    print(net)
 # 访问 https://www.jetbrains.com/help/pycharm/ 获取 PyCharm 帮助