[Feature] TensorDictCatView

ghstack-source-id: eda35393bab9de459fc01c6e33a872ffb1b1672a Pull Request resolved: #1037
pytorch · Oct 10, 2024 · 4a406ba · 4a406ba
1 parent 1e32195
commit 4a406ba
Show file tree

Hide file tree

Showing 5 changed files with 1,124 additions and 26 deletions.
diff --git a/tensordict/__init__.py b/tensordict/__init__.py
@@ -4,7 +4,7 @@
 # LICENSE file in the root directory of this source tree.
 
 import tensordict._reductions
-from tensordict._lazy import LazyStackedTensorDict
+from tensordict._lazy import LazyStackedTensorDict, TensorDictCatView
 from tensordict._nestedkey import NestedKey
 from tensordict._td import (
     cat,