def __init__(self, actor_net):
        """
        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

        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