Fonction torch.diff de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.diffest une fonction de PyTorch utilisée pour calculer les différences entre les éléments adjacents d’un tenseur. La formule de calcul est out[i] = input[i+1] - input[i].

Définition de fonction

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

Description des paramètres :

  • input: tenseur d'entrée
  • n: nombre de différences, la valeur par défaut est 1
  • dim: dimension le long de laquelle calculer les différences
  • prepend: ajouter en tête de l’entrée avant le calcul
  • append: ajouter à la fin de l’entrée après le calcul

Exemples d'utilisation

Exemple

import torch

# Créer un tenseur
x = torch.tensor([1, 3, 6, 10])

# Calculer la différence du premier ordre
y = torch.diff(x)
print(y)

Le résultat de sortie est:

tensor([2, 3, 4])

Exemple

import torch

# Créer un tenseur
x = torch.tensor([1, 3, 6, 10])

# Calculer la différence du second ordre
y = torch.diff(x, n=2)
print(y)

Le résultat de sortie est:

tensor([1, 1])

Exemple

import torch

# Créer un tenseur 2D
x = torch.tensor([[1, 3, 6, 10], [2, 5, 9, 14]])

# Calculer les différences le long de la dernière dimension
y = torch.diff(x)
print(y)

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions