nnrl/nn/model/stochastic/ensemble.py
def log_prob(self, new_obs: List[Tensor], params: List[TensorDict]) -> List[Tensor]:
        """Compute likelihoods for each model in the ensemble.

        Args:
            obs: List of `N` observation tensors of shape `(*,) + O`
            action: List of `N` action tensors of shape `(*,) + A`
            new_obs: List of `N` observation tensors of shape `(*,) + O`

        Returns:
           List of `N` log-likelihood tensors of shape `(*,)`
        """
        return [m.log_prob(new_obs[i], params[i]) for i, m in enumerate(self)]
Compute likelihoods for each model in the ensemble.

        Args:
            obs: List of `N` observation tensors of shape `(*,) + O`
            action: List of `N` action tensors of shape `(*,) + A`
            new_obs: List of `N` observation tensors of shape `(*,) + O`

        Returns:
           List of `N` log-likelihood tensors of shape `(*,)`
        
Compute likelihoods for each model in the ensemble.
Compute likelihoods for each model in the ensemble .
def log_prob(self, new_obs: List[Tensor], params: List[TensorDict]) -> List[Tensor]:
        return [m.log_prob(new_obs[i], params[i]) for i, m in enumerate(self)]
def log_prob ( self , new_obs : List [ Tensor ] , params : List [ TensorDict ] ) -> List [ Tensor ] : return [ m . log_prob ( new_obs [ i ] , params [ i ] ) for i , m in enumerate ( self ) ]
Compute likelihoods for each model in the ensemble.
Compute likelihoods for each model in the ensemble .
"""Compute likelihoods for each model in the ensemble.

        Args:
            obs: List of `N` observation tensors of shape `(*,) + O`
            action: List of `N` action tensors of shape `(*,) + A`
            new_obs: List of `N` observation tensors of shape `(*,) + O`

        Returns:
           List of `N` log-likelihood tensors of shape `(*,)`
        """
List of `N` log-likelihood tensors of shape `(*,)`
List of ` N ` log - likelihood tensors of shape ` ( * ) `
self
new_obs
List[Tensor]
List of `N` observation tensors of shape `(*,) + O`
List of ` N ` observation tensors of shape ` ( * ) + O `
params
List[TensorDict]
obs
List of `N` observation tensors of shape `(*,) + O`
List of ` N ` observation tensors of shape ` ( * ) + O `
action
List of `N` action tensors of shape `(*,) + A`
List of ` N ` action tensors of shape ` ( * ) + A `