torch.masked_select — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.masked_select. Returns a new 1-D tensor which indexes the input tensor according to the boolean mask mask which is a BoolTensor. The shapes of the mask tensor and the input tensor don’t need to match, but they must be broadcastable. input ( Tensor) – the input tensor. out ( Tensor, optional) – the output tensor.