maskedtensor Installation pip install maskedtensor You need to run a recent nightly of PyTorch. You can get it on the main website. Notebooks For an introduction and instructions on how to use MaskedTensors and what they are useful for see the notebooks folder.