def forward(self, x: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
        """Passes in a state x through the network and returns the policy and a sampled action.

        Args:
            x: environment state

        Returns:
            Tuple of policy and action

        """
        pi, action = self.actor(x)
        value = self.critic(x)

        return pi, action, value
Passes in a state x through the network and returns the policy and a sampled action.

        Args:
            x: environment state

        Returns:
            Tuple of policy and action

        
examples/pytorch/domain_templates/reinforce_learn_ppo.py