megengine.functional.svd

svd(inp, full_matrices=False, compute_uv=True)

对输入矩阵 inp 进行奇异值分解。

参数

inp (Tensor) – 输入矩阵，形状必须是 [..., M, N]。

返回类型

Tensor

返回

输出矩阵 (U, sigma, V)。

例如：

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor(np.arange(0, 6, dtype=np.float32).reshape(2,3))
_, y, _ = F.svd(x)
print(y.numpy().round(decimals=3))

输出：

[7.348 1.   ]

megengine.functional.matmul megengine.functional.var