human
(3) 정책경사를 위한 기반 기능 구현"
Please write code following the instructions in jupyter notebook style.

=====
Write me a code to (3) 정책경사를 위한 기반 기능 구현 and the other steps required?
assistant
#Here is code to (3) 정책경사를 위한 기반 기능 구현 and the other steps required.

class PGNET(nn.Module):
    # 순수하게 Policy gradient로 구성
    def __init__(self, num_state):
        super(PGNET, self).__init__()
    
        self.fc_in = nn.Linear(num_state, 24)
        self.fc_hidden = nn.Linear(24, 36)
        self.fc_out = nn.Linear(36, 1)
        
    def forward(self, x):
        x = F.relu(self.fc_in(x))
        x = F.relu(self.fc_hidden(x))
        x = TC.sigmoid(self.fc_out(x))
        return x