TensorDict is a pytorch dedicated tensor container. - View it on GitHub