models/policies.py
def output_layers(self, oshp):
        """
        Set the output layer needed for the distribution.

        Args:
            oshp: the input shape

        Return:
            layer: the corresponding layer
        """
        raise NotImplementedError

        Set the output layer needed for the distribution.

        Args:
            oshp: the input shape

        Return:
            layer: the corresponding layer
        
Set the output layer needed for the distribution.
Set the output layer needed for the distribution .
def output_layers(self, oshp):
        raise NotImplementedError
def output_layers ( self , oshp ) : raise NotImplementedError
Set the output layer needed for the distribution.
Set the output layer needed for the distribution .
"""
        Set the output layer needed for the distribution.

        Args:
            oshp: the input shape

        Return:
            layer: the corresponding layer
        """
self
oshp
the input shape
the input shape