Nested2TED

class torchrl.data.Nested2TED(done_key='done', shift_key='shift', is_full_key='is_full', done_keys=('done', 'truncated', 'terminated'), reward_keys=('reward',))

Converts a nested tensordict where each row is a trajectory into the TED format.