mlx.core.linalg.pinv

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

计算矩阵的（Moore-Penrose）伪逆。

此函数使用矩阵的奇异值分解计算其广义逆。此函数支持至少具有 2 个维度的数组。当输入具有多于两个维度时，将针对 a 的最后两个维度中的每个矩阵计算伪逆。

参数:

• a (array) – 输入数组。

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

返回:

aplus 满足 a @ aplus @ a = a

返回类型:

array