reference/generated/tensordict.nn.interaction_type

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

interaction_type

tensordict.nn.interaction_type() → InteractionType | None

Returns the interaction type set by set_interaction_type, or None outside of it.

When it returns None, each ProbabilisticTensorDictModule uses its default_interaction_type.

Examples

>>> from tensordict.nn import interaction_type, set_interaction_type
>>> print(interaction_type())
None
>>> with set_interaction_type("mode"):
...     print(interaction_type())
mode