dist_fn

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

---

class DistributionFunctionFactory

Bases: ToStringMixin, ABC

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

class DistributionFunctionFactoryCategorical(is_probs_input: bool = True)

Bases: DistributionFunctionFactory

Parameters:

is_probs_input – If True, the distribution function shall create a categorical distribution from a tensor containing probabilities; otherwise the tensor is assumed to contain logits.

create_dist_fn(envs: Environments) → Callable[[Tensor], Distribution]

class DistributionFunctionFactoryIndependentGaussians

Bases: DistributionFunctionFactory

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