Fonction torch.cumsum PyTorch

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

torch.cumsumest une fonction dans PyTorch utilisée pour calculer la somme cumulée. Elle renvoie la somme cumulée le long d'une dimension spécifiée, c'est-à-dire la somme de tous les éléments depuis le début jusqu'à la position actuelle.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

# Calculer la somme cumulée
x = torch.tensor([1, 2, 3, 4, 5])
result = torch.cumsum(x, dim=0)
print("Entrée :", x)
print("Somme cumulée :", result)
# Sortie : tensor([1, 3, 6, 10, 15])
# Explication : 1, 1+2=3, 1+2+3=6, 1+2+3+4=10, 1+2+3+4+5=15

# tenseur 2D
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
# Cumul le long des colonnes
result_col = torch.cumsum(x, dim=0)
print("\nSomme cumulative le long des colonnes:")
print(result_col)
# tensor([[ 1, 2, 3],
#         [ 5,  7,  9]])

# Cumul le long des lignes
result_row = torch.cumsum(x, dim=1)
print("\nSomme cumulative le long des lignes:")
print(result_row)
# tensor([[ 1, 3, 6],
#         [ 4,  9, 15]])

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

Autres extensions