Fonction PyTorch torch.linalg.matrix_power

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.linalg.matrix_powerC'est une fonction du module d'algèbre linéaire de PyTorch utilisée pour calculer les puissances de matrices. Elle calcule la n-ième puissance d'une matrice, A^n.

Définition de fonction

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

Parameters:

  • A(Tensor) : Matrice carrée d'entrée.
  • n(int): La puissance, qui peut être un entier, zéro ou un nombre négatif.
  • out(Tensor, optionnel) : Tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie la n-ième puissance de la matrice.

Exemples d'utilisation

Exemple

import torch

# Créer une matrice
A = torch.tensor([[1.0, 2.0],
                  [3.0, 4.0]])

# Calculer A à la puissance 2
A_power_2 = torch.linalg.matrix_power(A, 2)

print("Matrice A :")
print(A)
print("nA^2:")
print(A_power_2)
print("n vérification : A @ A =")
print(A @ A)

Le résultat de sortie est:

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

Exemple - A^0 et A^-1

import torch

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

# A^0 = I (matrice identité)
print("A^0:")
print(torch.linalg.matrix_power(A, 0))

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

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions