Función torch.diff de PyTorch

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

torch.diffEs una función en PyTorch utilizada para calcular la diferencia entre elementos adyacentes de un tensor. La fórmula de cálculo es out[i] = input[i+1] - input[i].

Definición de la función

torch.diff(input, n=1, dim=-1, prepend=None, append=None)

Descripción de parámetros:

  • input: tensor de entrada
  • n: número de diferencias, por defecto 1
  • dim: dimensión para calcular las diferencias
  • prepend: prepend (anteponer) a la entrada antes del cálculo
  • append: append (añadir) a la entrada después del cálculo

Ejemplo de uso

Ejemplo

import torch

# Crear tensor
x = torch.tensor([1, 3, 6, 10])

# Calcular la diferencia de primer orden
y = torch.diff(x)
print(y)

El resultado de salida es:

tensor([2, 3, 4])

Ejemplo

import torch

# Crear tensor
x = torch.tensor([1, 3, 6, 10])

# Calcular la diferencia de segundo orden
y = torch.diff(x, n=2)
print(y)

El resultado de salida es:

tensor([1, 1])

Ejemplo

import torch

# Crear un tensor bidimensional
x = torch.tensor([[1, 3, 6, 10], [2, 5, 9, 14]])

# Calcular la diferencia en la última dimensión
y = torch.diff(x)
print(y)

El resultado de salida es:

tensor([[2, 3, 4],
        [3, 4, 5]])

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

Otras extensiones