megengine.functional.svd

svd(inp, full_matrices=False, compute_uv=True)[源代码]

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

参数

inp – 输入矩阵,形状必须是 [..., M, N]

返回

输出矩阵 (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.   ]