Fonction torch.trace de PyTorch

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

torch.traceest une fonction de PyTorch utilisée pour calculer la trace d'une matrice. La trace est la somme des éléments de la diagonale principale d'une matrice.

Définition de fonction

torch.trace(input)

Description des paramètres :

  • input: tenseur d'entrée (au moins bidimensionnel)

Exemple d'utilisation

Exemple

import torch

# Créer une matrice
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Calculer la trace (somme des éléments de la diagonale principale)
y = torch.trace(a)
print(y)

Le résultat de sortie est:

tensor(15)

Exemple

import torch

# Créer une matrice non carrée
a = torch.tensor([[1, 2, 3], [4, 5, 6]])

# Calculer la trace
y = torch.trace(a)
print(y)

Le résultat de sortie est:

tensor(6)

Exemple

import torch

# Créer un tenseur multidimensionnel, ne prendre en compte que les deux premières dimensions
a = torch.randn(3, 4, 4, 5)
y = torch.trace(a)
print(y.shape)

Le résultat de sortie est:

torch.Size([3, 5])

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

Autres extensions