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.norm