Fonction torch.matrix_power de PyTorch


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

torch.matrix_powerest une fonction de PyTorch utilisée pour calculer les puissances matricielles.

Définition de fonction

torch.matrix_power(input, n)

Exemple d'utilisation

Exemple

import torch

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

# Calculer la puissance 3 de la matrice
A_power = torch.matrix_power(A, 3)
print("Puissance de la matrice :")
print(A_power)

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

Autres extensions