Функция PyTorch torch.trace

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.traceэто функция в PyTorch, используемая для вычисления следа матрицы. След — это сумма элементов на главной диагонали матрицы.

Определение функции

torch.trace(input)

Описание параметров:

  • input: входной тензор (как минимум двумерный)

Пример использования

Пример

import torch

# Создать матрицу
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Вычислить след (сумма элементов главной диагонали)
y = torch.trace(a)
print(y)

Результат вывода:

tensor(15)

Пример

import torch

# Создать неквадратную матрицу
a = torch.tensor([[1, 2, 3], [4, 5, 6]])

# Вычислить след
y = torch.trace(a)
print(y)

Результат вывода:

tensor(6)

Пример

import torch

# Создать многомерный тензор, учитывать только первые два измерения
a = torch.randn(3, 4, 4, 5)
y = torch.trace(a)
print(y.shape)

Результат вывода:

torch.Size([3, 5])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения