假設我們要求A的逆矩陣,正常情況下我們使用如下命令:
torch.inverse(A)
但是本人發現,這個函數還挺消耗顯存的。想到求逆矩陣和求線性方程組有很大關系。從而可以使用torch.linalg.solve來求解逆矩陣,關鍵是其顯存消耗小。
求解逆矩陣不就是求解下面的X嗎?
AX=IAX=IAX=I
所以我們可以調用torch.linalg.solve來求解。
torch.linalg.solve(A,torch.eye(len(A)))
反正可以節約蠻多顯存的,尤其是當A的維度比較高,比如上萬,精度比較高,比如float64,的時候。