Skip to main content

MaskedTensors for PyTorch

Project description

maskedtensor

Warning: This is a prototype library that is actively under development. If you have suggestions or potential use cases that you'd like addressed, please open a Github issue; we welcome any thoughts, feedback, and contributions!

MaskedTensor is a prototype library that is part of the PyTorch project and is an extension of torch.Tensor that provides the ability to mask out the value for any given element. Elements with masked out values are ignored during computation and give the user access to advanced semantics such as masked reductions, safe softmax, masked matrix multiplication, filtering NaNs, and masking out certain gradient values.

Installation

Binaries

To install the official MaskedTensor via pip, use the following command:

pip install maskedtensor

For the dev (unstable) nightly version that contains the most recent features, please replace maskedtensor with maskedtensor-nightly.

Note that MaskedTensor requires PyTorch >= 1.11, which you can get on the the main website

From Source

To install from source, you will need Python 3.7 or later, and we highly recommend that you use an Anaconda environment. Then run:

python setup.py develop

Documentation

Please find documentation on the MaskedTensor Website.

Building documentation

Please follow the instructions in the docs README.

Notebooks

For an introduction and instructions on how to use MaskedTensors and what they are useful for, there are a nubmer of tutorials on the MaskedTensor website.

License

maskedtensor is licensed under BSD 3-Clause

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

File details

Details for the file maskedtensor_nightly-0.11.dev2022618-py3-none-any.whl.

File metadata

File hashes

Hashes for maskedtensor_nightly-0.11.dev2022618-py3-none-any.whl
Algorithm Hash digest
SHA256 44528bfd62f3bc255ed79daf3ad5cb9d3f2836e36b7d6e31c92e85f95fe35938
MD5 b68967df4873caec15658a8d9ac8853e
BLAKE2b-256 bdf580e8545039188df3c3c118de8c2c98318bab67a6c1bdd0ab3002385e829c

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page