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

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

torch.triuЭто функция в PyTorch, используемая для извлечения верхней треугольной части матрицы (включая главную диагональ). Нижняя треугольная часть будет установлена в 0.

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

torch.triu(input, diagonal=0, out=None)

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

  • input: входной тензор
  • diagonal: индекс диагонали, 0 обозначает главную диагональ
  • out: выходной тензор

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

Пример

import torch

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

# Извлечь верхнюю треугольную часть
y = torch.triu(a)
print(y)

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

tensor([[1, 2, 3],
        [0, 5, 6],
        [0, 0, 9]])

Пример

import torch

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

# Извлечь верхнюю треугольную часть выше первой диагонали, расположенной над главной диагональю
y = torch.triu(a, diagonal=1)
print(y)

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

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

Пример

import torch

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

# Извлечь верхнюю треугольную часть
y = torch.triu(a)
print(y)

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

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

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

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