set_interaction_type

class tensordict.nn.set_interaction_type(type: tensordict.nn.probabilistic.InteractionType | str | None = InteractionType.DETERMINISTIC)

Sets all ProbabilisticTDModules sampling to the desired type.

Parameters:

type (InteractionType or str) – sampling type to use when the policy is being called.