2024年1月10日
实现自定义算子：逆矩阵inverse
摘要： 逆矩阵Cuda不支持，只能手动实现 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[in 阅读全文
posted @ 2024-01-10 21:01 jimchen1218 阅读(68) 评论(0) 推荐(0)