"invalid shape dimension <huge negative number>" on tensor masking operation

@Algomorph I too encountered this issue when loading a tensor from the disk and trying to filter it with a mask.
Your solution helped me as well, but it is indeed confusing why…

If anyone has any insights pls let us know.