ray.rllib.models.distributions.Distribution.get_partial_dist_cls

classmethod Distribution.get_partial_dist_cls(**partial_kwargs) → Distribution

Returns a partial child of TorchMultiActionDistribution.

This is useful if inputs needed to instantiate the Distribution from logits are available, but the logits are not.