dist_fn

Contents

dist_fn#

Source code: tianshou/highlevel/params/dist_fn.py

class DistributionFunctionFactory[source]#

abstract create_dist_fn(envs: Environments) → Callable[[Any], Distribution][source]#

class DistributionFunctionFactoryCategorical[source]#

create_dist_fn(envs: Environments) → Callable[[Tensor], Categorical][source]#

class DistributionFunctionFactoryDefault[source]#

create_dist_fn(envs: Environments) → Callable[[tuple[Tensor, Tensor]], Distribution] | Callable[[Tensor], Categorical][source]#

class DistributionFunctionFactoryIndependentGaussians[source]#

create_dist_fn(envs: Environments) → Callable[[tuple[Tensor, Tensor]], Distribution] | Callable[[Tensor], Categorical][source]#