Función torch.trace de PyTorch

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

torch.traceEs una función en PyTorch utilizada para calcular la traza de una matriz. La traza es la suma de los elementos de la diagonal principal de una matriz.

Definición de la función

torch.trace(input)

Descripción de parámetros:

  • input: Tensor de entrada (al menos bidimensional)

Ejemplo de uso

Ejemplo

import torch

# Crear la matriz
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Calcular la traza (suma de los elementos de la diagonal principal)
y = torch.trace(a)
print(y)

El resultado de salida es:

tensor(15)

Ejemplo

import torch

# Crear una matriz no cuadrada
a = torch.tensor([[1, 2, 3], [4, 5, 6]])

# Calcular la traza
y = torch.trace(a)
print(y)

El resultado de salida es:

tensor(6)

Ejemplo

import torch

# Crear un tensor multidimensional, teniendo en cuenta solo las dos primeras dimensiones
a = torch.randn(3, 4, 4, 5)
y = torch.trace(a)
print(y.shape)

El resultado de salida es:

torch.Size([3, 5])

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

Otras extensiones