Función PyTorch torch.svd

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.svdEs la función en PyTorch para calcular la descomposición en valores singulares de una matriz.

Definición de la función

torch.svd(input, some, compute_uv, out)

Ejemplo de uso

Ejemplo

import torch

A = torch.randn(3, 4)

# Descomposición SVD
U, S, V = torch.svd(A)

print("Forma de U:", U.shape)
print("Forma de S:", S.shape)
print("Forma de V:", V.shape)

El resultado de salida es:

U 形状: torch.Size([3, 3])
S 形状: torch.Size([3])
V 形状: torch.Size([4, 4])

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones