Función PyTorch torch.tril

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

torch.trilEs una función de PyTorch que se utiliza para extraer la parte triangular inferior de una matriz (incluida la diagonal principal). La parte triangular superior se establece en 0.

Definición de la función

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

Descripción de parámetros:

  • input: tensor de entrada
  • diagonal: índice de la diagonal, 0 indica la diagonal principal
  • out: tensor de salida

Ejemplo de uso

Ejemplo

import torch

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

# Extraer la parte triangular inferior
y = torch.tril(a)
print(y)

El resultado de salida es:

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

Ejemplo

import torch

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

# Extraer la parte triangular inferior desde la primera diagonal debajo de la diagonal principal
y = torch.tril(a, diagonal=1)
print(y)

El resultado de salida es:

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

Ejemplo

import torch

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

# Extraer la parte triangular inferior
y = torch.tril(a)
print(y)

El resultado de salida es:

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

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

Otras extensiones