Shortcuts

tensordict.nn.set_interaction_type

class tensordict.nn.set_interaction_type(type: InteractionType | None = InteractionType.MODE)

Sets all ProbabilisticTDModules sampling to the desired type.

Parameters:

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources