Update

[ghstack-poisoned]
pytorch · Feb 24, 2025 · cd454e1 · cd454e1
2 parents 26c9d7b + 970fb0e
commit cd454e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchrl/envs/common.py b/torchrl/envs/common.py
@@ -1967,6 +1967,7 @@ def select_and_clone(name, x, y):
             select_and_clone,
             next_tensordict,
             device=next_tensordict.device,
+            batch_size=next_tensordict.batch_size,
             default=None,
             filter_empty=True,
             is_leaf=_is_leaf_nontensor,