text: import torch.nn as nn
  import torch.nn.functional as F
  
  
  class SimpleNet(nn.Module):
      def __init__(self, state_dim, act_dim):
          super(SimpleNet, self).__init__()
          self.fc1 = nn.Linear(state_dim, 32)
          self.fc2 = nn.Linear(32, 32)
          self.fc3 = nn.Linear(32, act_dim)
  
      def forward(self, x):
          x = F.relu(self.fc1(x))
          x = F.relu(self.fc2(x))
          x = self.fc3(x)
          return x
  
  
  