.. currentmodule:: tensordict.nn WrapModule ========== .. autoclass:: WrapModule :members: