maskedtensor

MaskedTensor是一个库，它是PyTorch项目的一部分，通过提供对任何给定元素值的屏蔽能力，扩展了torch.Tensor。屏蔽了值的元素在计算时被忽略，并使用户能够访问高级语义，例如屏蔽 reductions、安全的 softmax、屏蔽矩阵乘法、过滤 NaNs 和屏蔽某些梯度值。

安装

二进制文件

要使用pip安装MaskedTensor，请使用以下命令

pip install maskedtensor

请注意，MaskedTensor需要PyTorch >= 1.11，您可以在官方网站上获取

从源安装

要从源安装，您将需要Python 3.7或更高版本，我们强烈建议您使用Anaconda环境。然后运行

python setup.py develop

文档

请参阅MaskedTensor网站上的文档。

构建文档

请按照docs README中的说明操作。

笔记本

有关MaskedTensors的介绍和使用说明以及它们有哪些用途，MaskedTensor网站上有一些教程。

许可证

maskedtensor受BSD 3-Clause许可证许可