[Feature] TensorDictCatView

ghstack-source-id: 672dc8b82c0b025feab98d061b5241536fa040c0 Pull Request resolved: #1037
pytorch · Oct 11, 2024 · fa3c693 · fa3c693
1 parent 1e32195
commit fa3c693
Show file tree

Hide file tree

Showing 5 changed files with 1,155 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,