Función PyTorch torch.triu

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

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

Definición de la función

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

Descripción de parámetros:

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

Ejemplos de uso

Ejemplo

import torch

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

# Extraer la parte triangular superior
y = torch.triu(a)
print(y)

El resultado de salida es:

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

Ejemplo

import torch

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

# Extraer el triángulo superior de la primera diagonal por encima de la diagonal principal
y = torch.triu(a, diagonal=1)
print(y)

El resultado de salida es:

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

Ejemplo

import torch

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

# Extraer la parte triangular superior
y = torch.triu(a)
print(y)

El resultado de salida es:

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

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

Otras extensiones