mlx.core.linalg.tri_inv

tri_inv(a: array, upper: bool = False, *, stream: None | Stream | Device = None) → array

计算三角方阵的逆。

此函数支持至少具有 2 个维度的数组。当输入具有两个以上的维度时，将对 a 的最后两个维度中的每个矩阵计算逆。

参数:

• a (array) – 输入数组。

• upper (bool, optional) – 数组是上三角还是下三角。默认为 False。

• stream (Stream, optional) – 流或设备。默认为 None，在这种情况下使用默认设备的默认流。

返回值:

ainv 满足 dot(a, ainv) = dot(ainv, a) = eye(a.shape[0])

返回类型:

array