Skip to content

Commit fa3c693

Browse files
author
Vincent Moens
committed
[Feature] TensorDictCatView
ghstack-source-id: 672dc8b Pull Request resolved: #1037
1 parent 1e32195 commit fa3c693

File tree

5 files changed

+1155
-26
lines changed

5 files changed

+1155
-26
lines changed

tensordict/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
# LICENSE file in the root directory of this source tree.
55

66
import tensordict._reductions
7-
from tensordict._lazy import LazyStackedTensorDict
7+
from tensordict._lazy import LazyStackedTensorDict, TensorDictCatView
88
from tensordict._nestedkey import NestedKey
99
from tensordict._td import (
1010
cat,

0 commit comments

Comments
 (0)