nnrl/nn/actor/policy/stochastic.py
def sample(self, obs, sample_shape: List[int] = ()):
        """
        Generates a sample_shape shaped sample or sample_shape shaped batch of
        samples if the distribution parameters are batched. Returns a (sample, log_prob)
        pair.
        """
        params = self(obs)
        return self.dist.sample(params, sample_shape)

        Generates a sample_shape shaped sample or sample_shape shaped batch of
        samples if the distribution parameters are batched. Returns a (sample, log_prob)
        pair.
        
Generates a sample_shape shaped sample or sample_shape shaped batch of
samples if the distribution parameters are batched. Returns a (sample, log_prob)
pair.
Generates a sample_shape shaped sample or sample_shape shaped batch of samples if the distribution parameters are batched . Returns a ( sample log_prob ) pair .
def sample(self, obs, sample_shape: List[int] = ()):
        params = self(obs)
        return self.dist.sample(params, sample_shape)
def sample ( self , obs , sample_shape : List [ int ] = ( ) ) : params = self ( obs ) return self . dist . sample ( params , sample_shape )
Generates a sample_shape shaped sample or sample_shape shaped batch of
samples if the distribution parameters are batched.
Generates a sample_shape shaped sample or sample_shape shaped batch of samples if the distribution parameters are batched .
"""
        Generates a sample_shape shaped sample or sample_shape shaped batch of
        samples if the distribution parameters are batched. Returns a (sample, log_prob)
        pair.
        """
self
obs
sample_shape
List[int]