Función torch.linalg.matrix_power de PyTorch

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

torch.linalg.matrix_powerEs una función del módulo de álgebra lineal de PyTorch que se utiliza para calcular la potencia de matrices. Calcula la n-ésima potencia de la matriz A^n.

Definición de la función

torch.linalg.matrix_power(A, n, out=None)

Parámetros:

  • A(Tensor): la matriz cuadrada de entrada.
  • n(int): la potencia, que puede ser un entero, cero o un número negativo.
  • out(Tensor, opcional): el tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve la n-ésima potencia de la matriz.

Ejemplo de uso

Ejemplo

import torch

# Crear la matriz
A = torch.tensor([[1.0, 2.0],
                  [3.0, 4.0]])

# Calcular la potencia 2 de A
A_power_2 = torch.linalg.matrix_power(A, 2)

print("Matriz A:")
print(A)
print("nA^2:")
print(A_power_2)
print("Verificación n: A @ A =")
print(A @ A)

El resultado de salida es:

矩阵 A:
tensor([[1., 2.],
        [3., 4.]])
A^2:
tensor([[ 7., 10.],
        [15., 22.]])
验证: A @ A =
tensor([[ 7., 10.],
        [15., 22.]])

Ejemplo - A^0 y A^-1

import torch

A = torch.tensor([[1.0, 2.0],
                  [3.0, 4.0]], dtype=torch.float64)

# A^0 = I (matriz identidad)
print("A^0:")
print(torch.linalg.matrix_power(A, 0))

# A^-1 = la matriz inversa de A
print("nA^-1:")
print(torch.linalg.matrix_power(A, -1))

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

Otras extensiones