Fonction torch.cumprod PyTorch

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

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

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

# Calculer le produit cumulé
x = torch.tensor([1, 2, 3, 4, 5])
result = torch.cumprod(x, dim=0)
print("Entrée :", x)
print("Produit cumulatif :", result)
# Sortie : tensor([1, 2, 6, 24, 120])
# Explication : 1, 1*2=2, 1*2*3=6, 1*2*3*4=24, 1*2*3*4*5=120

# tenseur 2D
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
# Cumul par colonne
result_col = torch.cumprod(x, dim=0)
print("nProduit cumulatif par colonne :")
print(result_col)
# tensor([[ 1, 2, 3],
#         [ 4, 10, 18]])

# Cumul par ligne
result_row = torch.cumprod(x, dim=1)
print("nProduit cumulatif par ligne :")
print(result_row)
# tensor([[ 1, 2, 6],
#         [  4,  20, 120]])

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

Autres extensions