跳转到主要内容

PyTorch的MaskedTensors

项目描述

maskedtensor

警告:这是一个正在积极开发的原型库。如果您有建议或希望解决的问题,请打开Github问题;我们欢迎任何想法、反馈和贡献!

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

安装

二进制文件

要使用pip安装官方的MaskedTensor,请使用以下命令

pip install maskedtensor

对于包含最新功能的开发(不稳定)夜间版本,请将maskedtensor替换为maskedtensor-nightly

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

从源码安装

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

python setup.py develop

文档

请查阅MaskedTensor网站上的文档。

构建文档

请遵循docs README中的说明。

笔记本

有关MaskedTensors的介绍和使用说明,以及它们有什么用,可以在MaskedTensor网站上找到一些教程。

许可证

maskedtensor采用BSD 3-Clause许可证

项目详情


下载文件

下载适用于您的平台的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源代码发行版

此版本没有可用的源代码发行版文件。请参阅生成发行版存档的教程。

构建发行版

maskedtensor_nightly-0.12.dev2022714-py3-none-any.whl (17.9 kB 查看哈希值)

上传时间 Python 3

支持者

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF 赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误记录 StatusPage StatusPage 状态页面