PyTorch función torch.cumsum

Pytorch torch 参考手册Pytorch manual de referencia de torch

torch.cumsumEs una función de PyTorch utilizada para calcular la suma acumulativa. Devuelve la suma acumulativa a lo largo de la dimensión especificada, es decir, la suma de todos los elementos desde el inicio hasta la posición actual.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Calcular la suma acumulativa
x = torch.tensor([1, 2, 3, 4, 5])
result = torch.cumsum(x, dim=0)
print("Entrada:", x)
print("Suma acumulativa:", result)
# Salida: tensor([1, 3, 6, 10, 15])
# Explicación: 1, 1+2=3, 1+2+3=6, 1+2+3+4=10, 1+2+3+4+5=15

# Tensor 2D
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
# Acumular por columna
result_col = torch.cumsum(x, dim=0)
print("nSuma acumulativa por columna:")
print(result_col)
# tensor([[ 1,  2,  3],
#         [ 5,  7,  9]])

# Acumular por fila
result_row = torch.cumsum(x, dim=1)
print("nSuma acumulativa por fila:")
print(result_row)
# tensor([[ 1,  3,  6],
#         [ 4,  9, 15]])

Pytorch torch 参考手册Pytorch manual de referencia de torch

Otras extensiones