dnn_6_f.py 文件源码

def Q_func(self, state):

        s = Variable(state)
        h = F.tanh(self.fc1(state))
        h = F.tanh(self.fc2(h))  
        h = F.tanh(self.fc3(h))
        h = F.tanh(self.fc4(h))
        h = F.tanh(self.fc5(h))
        Q = self.q_value(h)

        return Q