From 7b58cc30ed8c66820a819ebb7f4df1cbf258a11c Mon Sep 17 00:00:00 2001
From: Shion <26x.orc.ed5.1hs@gmail.com>
Date: Thu, 26 Aug 2021 09:21:23 +0900
Subject: [PATCH] fix rnn parameter

---
 dezero/layers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dezero/layers.py b/dezero/layers.py
index 564279f..7a76f9d 100644
--- a/dezero/layers.py
+++ b/dezero/layers.py
@@ -231,7 +231,7 @@ def __init__(self, hidden_size, in_size=None):
         """
         super().__init__()
         self.x2h = Linear(hidden_size, in_size=in_size)
-        self.h2h = Linear(hidden_size, in_size=in_size, nobias=True)
+        self.h2h = Linear(hidden_size, in_size=hidden_size, nobias=True)
         self.h = None
 
     def reset_state(self):
@@ -328,4 +328,4 @@ def __call__(self, x):
         if self.avg_mean.data is None:
             self._init_params(x)
         return F.batch_nrom(x, self.gamma, self.beta, self.avg_mean.data,
-                            self.avg_var.data)
\ No newline at end of file
+                            self.avg_var.data)