def __init__(self, actor_net, act_dim):
        """
        Args:
            input_shape: observation shape of the environment
            n_actions: number of discrete actions available in the environment
        """
        super().__init__()
        self.actor_net = actor_net
        log_std = -0.5 * torch.ones(act_dim, dtype=torch.float)
        self.log_std = nn.Parameter(log_std)

        Args:
            input_shape: observation shape of the environment
            n_actions: number of discrete actions available in the environment
        
examples/pytorch/domain_templates/reinforce_learn_ppo.py