torch::linalg::solve函数原型为
inline Tensor solve(const Tensor& input, const Tensor& other, bool left)
torch::Tensor AXinv = torch::linalg::solve(X, A, false); // A * Xinv
torch::Tensor XinvA = torch::linalg::solve(X, A, true); // Xinv * A
torch::Tensor AinvX = torch::linalg::solve(A, X, true); // Ainv * X
torch::Tensor XAinv = torch::linalg::solve(A, X, false); // X * Ainv
BOOL参数是指第一个参数的逆是在第二个参数的左边还是第二个参数的右边,左边是true, 右边是false
714

被折叠的 条评论
为什么被折叠?



