Skip to content

Commit a121330

Browse files
[Doc] Fix typo in pad_sequence docstring example (#1262)
1 parent f7bc839 commit a121330

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tensordict/functional.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -127,11 +127,11 @@ def pad_sequence(
127127
>>> print(padded_td)
128128
TensorDict(
129129
fields={
130-
a: Tensor(shape=torch.Size([2, 4, 8]), device=cpu, dtype=torch.float32, is_shared=False),
131-
b: Tensor(shape=torch.Size([2, 5, 8]), device=cpu, dtype=torch.float32, is_shared=False),
130+
a: Tensor(shape=torch.Size([2, 5, 8]), device=cpu, dtype=torch.float32, is_shared=False),
131+
b: Tensor(shape=torch.Size([2, 6, 8]), device=cpu, dtype=torch.float32, is_shared=False),
132132
masks: TensorDict(
133133
fields={
134-
a: Tensor(shape=torch.Size([2, 4]), device=cpu, dtype=torch.bool, is_shared=False),
134+
a: Tensor(shape=torch.Size([2, 5]), device=cpu, dtype=torch.bool, is_shared=False),
135135
b: Tensor(shape=torch.Size([2, 6]), device=cpu, dtype=torch.bool, is_shared=False)},
136136
batch_size=torch.Size([2]),
137137
device=None,

0 commit comments

Comments
 (0)