Функция torch.diagflat в PyTorch

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

torch.diagflatЭто функция в PyTorch, используемая для создания диагональной плоской матрицы. Независимо от того, является ли входной тензор одномерным или многомерным, он будет уплощен и использован в качестве диагональных элементов.

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

torch.diagflat(input, diagonal=0)

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

  • input: Входной тензор, будет уплощен
  • diagonal: Индекс диагонали

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

Пример

import torch

# Создать одномерный тензор
x = torch.tensor([1, 2, 3])

# Создать диагональную плоскую матрицу
y = torch.diagflat(x)
print(y)

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

tensor([[1, 0, 0],
        [0, 2, 0],
        [0, 0, 3]])

Пример

import torch

# Создать двумерный тензор
x = torch.tensor([[1, 2], [3, 4]])

# Будет уплощен до [1,2,3,4], создавая диагональную матрицу 4x4
y = torch.diagflat(x)
print(y)

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

tensor([[1, 0, 0, 0],
        [0, 2, 0, 0],
        [0, 0, 3, 0],
        [0, 0, 0, 4]])

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

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