[FEATURE] TensorDictModule in_keys allowed as Dict[str, tuple | list] to enable multi use of a sample feature#1101

Merged

vmoens merged 1 commit intopytorch:mainfrom bachdj-px:support_tuple_in_keys_dispatch

Nov 29, 2024

+131-8

Commits on Nov 28, 2024

feat(nn.common.TensorDictModule): support tuple values in in_keys for flexible input key dispatching
bachdj-px
committed