bytetensor于longtensor对于索引的影响

import torch

boxes = torch.randint(6, (3,4))
boxes

#        [1., 4., 5., 2.],
#        [5., 5., 1., 5.]])
mask1 = torch.tensor([0,1,0], dtype=torch.long).view(-1,1)
mask2 = torch.tensor([0,1,0], dtype=torch.uint8).view(-1,1)

boxes[mask1.expand_as(boxes)]
#tensor([[[0., 4., 1., 1.],
#         [0., 4., 1., 1.],
#         [0., 4., 1., 1.],
#         [0., 4., 1., 1.]],

#        [[1., 4., 5., 2.],
#         [1., 4., 5., 2.],
#         [1., 4., 5., 2.],
#         [1., 4., 5., 2.]],

#        [[0., 4., 1., 1.],
#         [0., 4., 1., 1.],
#         [0., 4., 1., 1.],
#         [0., 4., 1., 1.]]])

boxes[mask2.expand_as(boxes)]
#tensor([1., 4., 5., 2.])