PyTorch torch.cumprod función
Pytorch torch manual de referencia
torch.cumprodEs una función en PyTorch que se utiliza para calcular el producto acumulativo. Devuelve el producto acumulado a lo largo de la dimensión especificada, es decir, el producto de todos los elementos desde el inicio hasta la posición actual.
Definición de función
torch.cumprod(input, dim, dtype=None)
Ejemplo de uso
Ejemplo
import torch
# Calcular el producto acumulativo
x = torch.tensor([1, 2, 3, 4, 5])
result = torch.cumprod(x, dim=0)
print("Entrada:", x)
print("Producto acumulativo:", result)
# Salida: tensor([1, 2, 6, 24, 120])
# Explicación: 1, 1*2=2, 1*2*3=6, 1*2*3*4=24, 1*2*3*4*5=120
# Tensor 2D
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
# Acumular por columnas
result_col = torch.cumprod(x, dim=0)
print("nProducto acumulativo por columnas:")
print(result_col)
# tensor([[ 1, 2, 3],
# [ 4, 10, 18]])
# Acumular por filas
result_row = torch.cumprod(x, dim=1)
print("nProducto acumulativo por filas:")
print(result_row)
# tensor([[ 1, 2, 6],
# [ 4, 20, 120]])
# Calcular el producto acumulativo
x = torch.tensor([1, 2, 3, 4, 5])
result = torch.cumprod(x, dim=0)
print("Entrada:", x)
print("Producto acumulativo:", result)
# Salida: tensor([1, 2, 6, 24, 120])
# Explicación: 1, 1*2=2, 1*2*3=6, 1*2*3*4=24, 1*2*3*4*5=120
# Tensor 2D
x = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
# Acumular por columnas
result_col = torch.cumprod(x, dim=0)
print("nProducto acumulativo por columnas:")
print(result_col)
# tensor([[ 1, 2, 3],
# [ 4, 10, 18]])
# Acumular por filas
result_row = torch.cumprod(x, dim=1)
print("nProducto acumulativo por filas:")
print(result_row)
# tensor([[ 1, 2, 6],
# [ 4, 20, 120]])
Otras extensiones