PyTorch torch.diff-Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.diffist eine Funktion in PyTorch, die verwendet wird, um die Differenzen zwischen benachbarten Elementen eines Tensors zu berechnen. Die Berechnungsformel lautet out[i] = input[i+1] - input[i].

Funktionsdefinition

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

Parameterbeschreibung:

  • input: Eingabetensor
  • n: Anzahl der Differenzen, standardmäßig 1
  • dim: Dimension, entlang derer die Differenzen berechnet werden
  • prepend: vor der Berechnung an input voranstellen
  • append: nach der Berechnung an input anhängen

Anwendungsbeispiele

Beispiel

import torch

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

# Differenz erster Ordnung berechnen
y = torch.diff(x)
print(y)

Das Ausgabeergebnis ist:

tensor([2, 3, 4])

Beispiel

import torch

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

# Differenz zweiter Ordnung berechnen
y = torch.diff(x, n=2)
print(y)

Das Ausgabeergebnis ist:

tensor([1, 1])

Beispiel

import torch

# Einen 2D-Tensor erstellen
x = torch.tensor([[1, 3, 6, 10], [2, 5, 9, 14]])

# Differenzen entlang der letzten Dimension berechnen
y = torch.diff(x)
print(y)

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen