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许可证
项目详情
关闭
哈希值 for maskedtensor_nightly-0.12.dev2022714-py3-none-any.whl
算法 | 哈希摘要 | |
---|---|---|
SHA256 | 10fc6d015c6b149fd62f289c0b4a1bfb4831066f26099bd60b9a6ea03d42ccf9 |
|
MD5 | 99e17edf6a0b13a5a8c48cac90138ee2 |
|
BLAKE2b-256 | a59338c014dd8f84bfef60487a883432094c1cf2e24c05eaed068fb84967bec1 |