PyTorch torch.cummax-Funktion
PyTorch torch-Referenzhandbuch
torch.cummaxEs ist eine Funktion in PyTorch, die zur Berechnung kumulativer Maxima verwendet wird. Sie gibt das kumulative Maximum entlang einer angegebenen Dimension zusammen mit den entsprechenden Indizes zurück.
Funktionsdefinition
torch.cummax(input, dim, dtype=None)
Anwendungsbeispiel
Beispiel
import torch
# Kumulatives Maximum berechnen
x = torch.tensor([3, 1, 4, 1, 5, 9, 2, 6])
values, indices = torch.cummax(x, dim=0)
print("Eingabe:", x)
print("Kumulatives Maximum:", values)
print("Maximale Indizes:", indices)
# 2D-Tensor
x = torch.tensor([[3, 1, 4], [1, 5, 9], [2, 6, 5]], dtype=torch.float32)
# Spaltenweise akkumulieren
values_col, indices_col = torch.cummax(x, dim=0)
print("nSpaltenweise akkumulieren:")
print("Kumulatives Maximum:")
print(values_col)
print("Indizes:")
print(indices_col)
# Zeilenweise akkumulieren
values_row, indices_row = torch.cummax(x, dim=1)
print("nZeilenweise akkumulieren:")
print("Kumulatives Maximum:")
print(values_row)
# Kumulatives Maximum berechnen
x = torch.tensor([3, 1, 4, 1, 5, 9, 2, 6])
values, indices = torch.cummax(x, dim=0)
print("Eingabe:", x)
print("Kumulatives Maximum:", values)
print("Maximale Indizes:", indices)
# 2D-Tensor
x = torch.tensor([[3, 1, 4], [1, 5, 9], [2, 6, 5]], dtype=torch.float32)
# Spaltenweise akkumulieren
values_col, indices_col = torch.cummax(x, dim=0)
print("nSpaltenweise akkumulieren:")
print("Kumulatives Maximum:")
print(values_col)
print("Indizes:")
print(indices_col)
# Zeilenweise akkumulieren
values_row, indices_row = torch.cummax(x, dim=1)
print("nZeilenweise akkumulieren:")
print("Kumulatives Maximum:")
print(values_row)
Andere Erweiterungen