Función PyTorch torch.lu

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

torch.luEs una función de PyTorch que se utiliza para calcular la descomposición LU de una matriz. La descomposición LU descompone la matriz A en A = P * L * U, donde L es una matriz triangular inferior, U es una matriz triangular superior y P es una matriz de permutación.

Definición de la función

torch.lu(A, pivot=True, get_infos=False, out=None)

Parámetros:

  • A(Tensor): la matriz de entrada.
  • pivot(bool, opcional): si se realiza la descomposición LU con pivoteo por columnas. El valor predeterminado es True.
  • get_infos(bool, opcional): si es True, devuelve información. El valor predeterminado es False.
  • out(tuple, opcional): tupla de salida.

Valor de retorno:

  • tuple: devuelve una tupla (pivot, L, U).

Ejemplo de uso

Ejemplo

import torch

# Crear una matriz
A = torch.tensor([[1.0, 2.0, 3.0],
                  [4.0, 5.0, 6.0],
                  [7.0, 8.0, 9.0]])

# Descomposición LU
LU, pivots = torch.lu(A, pivot=True)

print("Matriz A:")
print(A)
print("nMatriz LU:")
print(LU)
print("nÍndice de pivote:")
print(pivots)

El resultado de salida es:

矩阵 A:
tensor([[1., 2., 3.],
        [4., 5., 6.],
        [7., 8., 9.]])
LU 矩阵:
tensor([[7., 8., 9.],
        [0.2, 0.4, 0.6],
        [0.6, 0.8, 0.0]])
主元索引:
tensor([3, 3, 3], dtype=torch.int32)

Ejemplo - usando get_infos

import torch

A = torch.tensor([[1.0, 2.0, 3.0],
                  [4.0, 5.0, 6.0],
                  [7.0, 8.0, 9.0]])

# Descomposición LU, con información
LU, pivots, info = torch.lu(A, pivot=True, get_infos=True)

print("Información:", info)

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

Otras extensiones