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

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

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

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

torch.tril(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.tril(a)
print(y)

Вывод:

tensor([[1, 0, 0],
        [4, 5, 0],
        [7, 8, 9]])

Пример

import torch

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

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

Вывод:

tensor([[1, 2, 0],
        [4, 5, 6],
        [7, 8, 9]])

Пример

import torch

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

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

Вывод:

tensor([[1, 0, 0],
        [4, 5, 0]])

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

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