Функция torch.svd в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.svdЭто функция в PyTorch, используемая для вычисления сингулярного разложения матрицы.

Определение функции

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

Пример использования

Пример

import torch

A = torch.randn(3, 4)

# SVD-разложение
U, S, V = torch.svd(A)

print("Форма U:", U.shape)
print("Форма S:", S.shape)
print("Форма V:", V.shape)

Вывод:

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

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения