Fonction torch.cummax de PyTorch

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

torch.cummaxIl s'agit d'une fonction de PyTorch utilisée pour calculer les maxima cumulés. Elle renvoie le maximum cumulé le long d'une dimension spécifiée ainsi que les indices correspondants.

Définition de fonction

torch.cummax(input, dim, dtype=None)

Exemple d'utilisation

Exemple

import torch

# Calculer le maximum cumulé
x = torch.tensor([3, 1, 4, 1, 5, 9, 2, 6])
valeurs, indices = torch.cummax(x, dim=0)
print("Entrée :", x)
print("Maximum cumulatif :", valeurs)
print("Indices des maximums :", indices)

# Tenseur 2D
x = torch.tensor([[3, 1, 4], [1, 5, 9], [2, 6, 5]], dtype=torch.float32)
# Accumuler par colonne
values_col, indices_col = torch.cummax(x, dim=0)
print("nAccumuler par colonne :")
print("Maximum cumulé :")
print(values_col)
print("Indices :")
print(indices_col)

# Accumuler par ligne
values_row, indices_row = torch.cummax(x, dim=1)
print("nAccumuler par ligne :")
print("Maximum cumulé :")
print(values_row)

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

Autres extensions