yolov1代码中的coor_mask.unsqueeze(-1) & coor_mask.expand_as(tensor)
importtorchtensor=torch.randn(1,7,7,30)print(tensor)print(tensor.shape)torch.Size([1,7,7,30])coor_mask=tensor[:,:,:,4]>1print(coor_mask)print(coor_mask.shape)tensor([[[True,False,True,False,True,False