torch.Tensor.masked_select

Tensor.masked_select(mask) → Tensor

参见 torch.masked_select()