torch-householder:PyTorch中的有效Householder转换

时间:2024-05-26 21:09:45
【文件属性】:

文件名称:torch-householder:PyTorch中的有效Householder转换

文件大小:176KB

文件格式:ZIP

更新时间:2024-05-26 21:09:45

Python

PyTorch中的有效住户转型 该存储库实现了Householder变换算法,用于计算正交矩阵和正交框架。 该算法以Python包的形式实现,并具有与PyTorch的不同绑定。 特别是,该软件包为torch.orgqr函数提供了增强的直接替换功能。 概述 从LAPACK开始,用于正交变换的API就已经存在。 但是,他们缺乏对深度学习框架的支持。 近日,正交约束已经在深度学习已成为流行的方式来规范模式,提高培训力度[ , ],因此通过arised正交变换需要backpropagate。 PyTorch 1.7实现矩阵指数函数torch.matrix_exp ,当输入矩阵倾斜对称时,可以将其重新用于执行正交变换。 这是我们在和评估中使用的基准。 与torch.matrix_exp相比,此程序包中实现的Householder转换具有以下优点: 数量级降低内存占用量 能够转换非平方矩阵


【文件预览】:
torch-householder-master
----MANIFEST.in(160B)
----LICENSE_code(2KB)
----requirements.txt(11B)
----doc()
--------img()
----torch_householder()
--------__init__.py(48B)
--------householder.cpp(4KB)
--------householder.py(3KB)
----setup.py(2KB)
----README.md(7KB)
----tests()
--------test.py(10KB)
--------requirements_test.txt(32B)
--------benchmark.py(9KB)
--------reference_orgqr.py(2KB)
----.gitignore(42B)
----package.sh(214B)
----pyproject.toml(120B)
----LICENSE_doc(18KB)

网友评论