masked_select(input, mask, *, out=None) → Tensor¶
Returns a new 1-D tensor which indexes the
inputtensor according to the boolean mask
maskwhich is a BoolTensor.
The shapes of the
masktensor and the
inputtensor don’t need to match, but they must be broadcastable.
The returned tensor does not use the same storage as the original tensor
- Keyword Arguments
out (Tensor, optional) – the output tensor.
>>> x = torch.randn(3, 4) >>> x tensor([[ 0.3552, -2.3825, -0.8297, 0.3477], [-1.2035, 1.2252, 0.5002, 0.6248], [ 0.1307, -2.0608, 0.1244, 2.0139]]) >>> mask = x.ge(0.5) >>> mask tensor([[False, False, False, False], [False, True, True, True], [False, False, False, True]]) >>> torch.masked_select(x, mask) tensor([ 1.2252, 0.5002, 0.6248, 2.0139])