PyTorch torch.matrix_power función


Pytorch torch 参考手册Pytorch torch manual de referencia

torch.matrix_powerEs una función en PyTorch para calcular la potencia de matrices.

Definición de la función

torch.matrix_power(input, n)

Ejemplo de uso

Ejemplo

import torch

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

# Calcular la potencia 3 de la matriz
A_power = torch.matrix_power(A, 3)
print("Potencia de la matriz:")
print(A_power)

Pytorch torch 参考手册Pytorch torch manual de referencia

Otras extensiones