You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
Repository navigation
[BUG] NonTensorStack.reshape gives the wrong batch size for a stack with a zero-size dim #2415
Cause, not traced to the line: NonTensorStack.reshape (tensordict/tensorclass.py:6396) reshapes a grid of member positions, which is empty here, and regroups the stack with _view or _select_positions; one of them keeps the old outer shape when there are no positions.
Found while reviewing #2410, which sends view of a zero-size lazy stack to reshape; for NonTensorStack it makes view give the same wrong sizes.
tensordict main (924b72a), torch 2.14.1, Python 3.12.12.
Activity