Función PyTorch torch.triangular_solve

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

torch.triangular_solveEs una función de PyTorch utilizada para resolver sistemas de ecuaciones lineales triangulares. Esta función resuelve AX = B, donde A es una matriz triangular.

Definición de la función

torch.triangular_solve(A, B, upper, transpose, unitriangular)

Descripción de parámetros

  • A: Matriz de coeficientes (debe ser una matriz cuadrada)
  • B: Matriz o vector del lado derecho
  • upper: Si A es una matriz triangular superior (por defecto True)
  • transpose: Si se transpone A (por defecto False)
  • unitriangular: Si se usa una triangular unitaria (por defecto False)

Ejemplo de uso

Ejemplo

import torch

# Crear la matriz de coeficientes triangular superior
A = torch.tensor([[3.0, 1.0, 2.0],
                  [0.0, 2.0, 1.0],
                  [0.0, 0.0, 1.0]])

# Vector del lado derecho
B = torch.tensor([9.0, 5.0, 2.0])

# Resolver AX = B
X = torch.triangular_solve(B.unsqueeze(1), A, upper=True)

print("Solución X:")
print(X.solution)

El resultado de salida es:

解 X:
tensor([[1.],
        [2.],
        [2.]])

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

Otras extensiones