From 44097da7288042e988bcb89f1c6cc817a8e1eec9 Mon Sep 17 00:00:00 2001
From: zhang <zch921005@126.com>
Date: Sat, 4 Jun 2022 08:48:54 +0800
Subject: 0604

---
 learn_torch/seq/test_rnn.py | 8 ++++++++
 1 file changed, 8 insertions(+)
 create mode 100644 learn_torch/seq/test_rnn.py

(limited to 'learn_torch/seq/test_rnn.py')

diff --git a/learn_torch/seq/test_rnn.py b/learn_torch/seq/test_rnn.py
new file mode 100644
index 0000000..5a7baf2
--- /dev/null
+++ b/learn_torch/seq/test_rnn.py
@@ -0,0 +1,8 @@
+import torch
+from torch import nn
+
+if __name__ == '__main__':
+    rnn = nn.RNN(10, 20, 2)
+    input = torch.randn(5, 3, 10)
+    h0 = torch.randn(2, 3, 10)
+    rnn(input, h0)
-- 
cgit v1.2.3