コード例 #1
0
 def __init__(self, hidden_size, out_size):
     super().__init__()
     self.rnn = L.RNN(hidden_size)
     self.h2y = L.Linear(out_size)
コード例 #2
0
if '__file__' in globals():
    import os, sys
    sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
import numpy as np
from dezero import Model
import dezero.functions as F
import dezero.layers as L
from dezero.utils import plot_dot_graph

rnn = L.RNN(10)  # 隠れ層のサイズだけを指定
x = np.random.rand(1, 1)
h = rnn(x)
print(h.shape)


class SimpleRNN(Model):
    def __init__(self, hidden_size, out_size):
        super().__init__()
        self.rnn = L.RNN(hidden_size)
        self.fc = L.Linear(out_size)

    def reset_state(self):
        self.rnn.reset_state()

    def forward(self, x):
        h = self.rnn(x)
        y = self.fc(h)
        return y


seq_data = [np.random.randn(1, 1) for _ in range(10)]  # ダミーの時系列データ