摘要: triangular_solve 算子简介 目标:求解上三角或者下三角作为系数方阵的线性系统,即求解 $AX=B$ 或者 $XA=B$,求解具有唯一解的 X。 接口输出参数: A:形状 [*, M, M],A 是一个方阵。其中 * 表示 batch_size,或者直接没有也行。 B:形状 [*, M 阅读全文
posted @ 2023-02-26 17:37 楷哥 阅读(171) 评论(0) 推荐(0) 编辑