mlx.core.linalg.inv

inv(a: array, *, stream: None | Stream | Device = None) → array

计算方阵的逆。

此函数支持至少二维的数组。当输入具有超过二维时，将在 a 的最后两个维度上为每个矩阵计算逆。

参数：

• a (array) – 输入数组。

• stream (Stream, optional) – 流或设备。默认为 None，此时使用默认设备的默认流。

返回：

ainv，使得 dot(a, ainv) = dot(ainv, a) = eye(a.shape[0])

返回类型：

array