pytorch实现矩阵的逆

拼搏现实的明天。 2022-11-15 11:35 221阅读 0赞

背景：本来pytorch是实现了逆矩阵的函数的，1.8之前是torch.inverse()，1.8是在torch.linalg.inv()

反正就是实现了，按理来说不需要去手动实现这玩意

但是，cuda不支持逆矩阵inverse，因此onnx当然也没实现inverse

但是我们训练的模型需要移植到onnx甚至cuda就会变得很困难，甚至你想在pytorch里面使用model.half()进行半精度运算的时候，就会报错说，cuda不支持inverse

因此，只能换个思路，自己实现这个算子，使用更为常见的算子来进行替代inverse

pytorch和numpy的源码很难找到inverse的具体实现，因此我只能另想办法，直到看到[这篇文章][Link 1]，人家使用numpy进行实现的，那我可以直接改为pytorch版的

import torch
    from torch.linalg import det
    
    def cof1(M,index):
        zs = M[:index[0]-1,:index[1]-1]
        ys = M[:index[0]-1,index[1]:]
        zx = M[index[0]:,:index[1]-1]
        yx = M[index[0]:,index[1]:]
        s = torch.cat((zs,ys),axis=1)
        x = torch.cat((zx,yx),axis=1)
        return det(torch.cat((s,x),axis=0))
    
    def alcof(M,index):
        return pow(-1,index[0]+index[1])*cof1(M,index)
    
    def adj(M):
        result = torch.zeros((M.shape[0],M.shape[1]))
        for i in range(1,M.shape[0]+1):
            for j in range(1,M.shape[1]+1):
                result[j-1][i-1] = alcof(M,[i,j])
        return result
    
    def invmat(M):
        return 1.0/det(M)*adj(M)
    
    M = torch.FloatTensor([[1,2,-1],[2,3,4],[3,1,2]])
    print(invmat(M))
    print(torch.inverse(M))

运行结果：

![2021033117244182.png][]

nice，搞定了

[Link 1]: https://www.pythonf.cn/read/54246
[2021033117244182.png]: /images/20221022/a554aee4f13f4d36bf826a9f36e53834.png

发表评论取消回复

表情：

评论列表（有 0 条评论，221人围观）

还没有评论，来说两句吧...

相关阅读

相关 17. 线性代数 - 矩阵的逆

文章目录矩阵的转置矩阵的逆 ![在这里插入图片描述][2c21bd062f9c47d0a0841a0c8a697508.png]

曾经终败给现在/ 2024年03月03日 08:10/ 0 赞/ 68 阅读

相关求解矩阵的逆矩阵方法

求解矩阵的逆矩阵方法矩阵的逆矩阵是线性代数中的一个重要概念，它在诸多实际问题中都有广泛的应用。在MATLAB中，我们可以使用内置的函数来计算矩阵的逆矩阵。本文将介绍两种常用

向右看齐/ 2023年10月15日 22:42/ 0 赞/ 15 阅读

相关牛顿法求解动态矩阵的逆、动态矩阵Moore-Penrose逆、动态矩阵形式线性方程问题

1. 牛顿法求解动态矩阵形式线性方程问题动态矩阵线性方程问题的定义：![A(t)x(t)=b(t)][A_t_x_t_b_t] 构造误差函数：![\\epsilon(

爱被打了一巴掌/ 2022年12月26日 00:49/ 0 赞/ 179 阅读

相关【线性代数（8）】矩阵行列式、伴随矩阵、逆矩阵

逆矩阵 1 矩阵行列式 2 伴随矩阵 3 逆矩阵 3.1 逆矩阵概念 3.2 逆矩阵的性质手动反爬虫： [原博地址]

Myth丶恋晨/ 2022年12月19日 08:35/ 0 赞/ 295 阅读

相关矩阵求逆c语言程序_用C和C ++程序查找矩阵的逆

![285e3e74ed8a0f07099fe012352e6edd.png][] 矩阵求逆c语言程序 Here you will get C and C++ progra

素颜马尾好姑娘i/ 2022年12月07日 14:00/ 0 赞/ 228 阅读

相关位姿矩阵求逆

位姿矩阵求逆位姿矩阵分析位姿矩阵求逆矩阵为了更好的说明问题，我们先来看一下位姿矩阵的定义。位姿矩阵分析如下图所示：如果在B坐标

╰+攻爆jí腚メ/ 2022年11月26日 08:53/ 0 赞/ 395 阅读

相关 pytorch实现矩阵的逆

背景：本来pytorch是实现了逆矩阵的函数的，1.8之前是torch.inverse()，1.8是在torch.linalg.inv() 反正就是实现了，按理来说不需要去手

拼搏现实的明天。/ 2022年11月15日 11:35/ 0 赞/ 222 阅读

相关数学_矩阵伪逆计算

直接上公式：A是（m,n）矩阵，(简单理解：m是行，代表方程个数；n代表列，表示未知数个数)。当A满秩时： ![20140502211056046][] 摘自： http:

水深无声/ 2022年10月22日 05:22/ 0 赞/ 180 阅读

相关矩阵的求逆

最近做一个加密算法遇到需要计算矩阵的逆，闲着无聊，记录一下，以后免得再麻烦。 include <stdio.h> include <mat

我会带着你远行/ 2022年09月18日 10:58/ 0 赞/ 244 阅读

相关 python求逆矩阵

方法1： from numpy import import numpy as np A=np.array([[1,2],[3,4

ゝ一世哀愁。/ 2022年09月13日 11:20/ 0 赞/ 217 阅读