- See [here](https://github.com/pytorch/pytorch/blob/4e3022dbe94af5db41a151fc7c2cbf717c583d0f/aten/src/ATen/native/native_functions.yaml#L9164-L9170) for operator schema - See [here](https://pytorch.org/docs/stable/generated/torch.masked_select.html) for `torch.masked_select` documentation