def forward(self, inputs, hidden_state):
    mb_size = inputs.size(0)
    x = F.relu(self.fc1(inputs))
    h_in = hidden_state.reshape((- 1), self.args.rnn_hidden_dim)
    h = self.rnn(x, h_in)
    if (not self.args.levin_flag_quantile):
        q = self.fc2(h)
    else:
        q = self.fc2(h).view(mb_size, self.args.n_actions, self.args.N_QUANT)
    return (q, h)
def forward(self, inputs, hidden_state):
    mb_size = inputs.size(0)
    x = F.relu(self.fc1(inputs))
    h_in = hidden_state.reshape((- 1), self.args.rnn_hidden_dim)
    h = self.rnn(x, h_in)
    if (not self.args.levin_flag_quantile):
        q = self.fc2(h)
    else:
        q = self.fc2(h).view(mb_size, self.args.n_actions, self.args.N_QUANT)
    return (q, h)