[BugFix] Compatibility of tensordict primers with batched envs (specifically for LSTM and GRU)

[BugFix] Compatibility of tensordict primers with batched envs (specifically for LSTM and GRU) #4096