Skip to content

Commit 0ee9e01

Browse files
Updates
1 parent 7c0d901 commit 0ee9e01

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torchrl/data/tensor_specs.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6826,6 +6826,7 @@ def _stack_specs(list_of_spec, dim=0, out=None):
68266826
else:
68276827
raise NotImplementedError
68286828

6829+
68296830
@TensorSpec.implements_for_spec(torch.index_select)
68306831
@Composite.implements_for_spec(torch.index_select)
68316832
def _index_select_spec(input: TensorSpec, dim: int, index: torch.Tensor) -> TensorSpec:

0 commit comments

Comments
 (0)