文件名称：数值实验代码matlab代码-matrix-sqrt:矩阵平方根及其梯度
文件大小：8KB
文件格式：ZIP
更新时间：2024-06-10 09:05:02
系统开源
数值实验代码matlab代码矩阵平方根及其梯度 概述 该存储库包含Python和Matlab代码，用于使用GPU上的各种技术来计算矩阵平方根（ZZ = A）及其梯度。 对于正向计算（矩阵的平方根），实现了SVD和迭代方法。 对于反向计算（平方根的梯度）矩阵后向，实现了使用SVD的Lyapunov求解器，通过正向迭代方法进行自动缩放以及使用迭代方法进行Lyapunov求解器。 这些实现提供了不同的数值精度，内存开销和速度折衷。 请参阅以下python代码的输出作为示例。 通常，迭代前进和后退方法的批处理模式版本比基于SVD的方法快一个数量级。 我们在“改进的双线性池”论文中使用了这些方法。 如果您使用以下代码，请引用此代码： @inproceedings{lin17improved, author = {Tsung-Yu Lin, and Subhransu Maji}, booktitle = {British Machine Vision Conference (BMVC)}, title = {Improved Bilinear Pooling with CNNs}, ye
【文件预览】：
matrix-sqrt-master
----matrix_sqrt.py(7KB)
----sqrt_backward.m(1KB)
----LICENSE(1KB)
----.gitignore(2B)
----lyap2.m(6KB)
----sqrt_forward.m(869B)
----README.md(4KB)
----matrix_sqrt.m(3KB)