Función torch.cummin de PyTorch

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

torch.cumminEs una función en PyTorch utilizada para calcular el mínimo acumulativo. Devuelve el mínimo acumulativo a lo largo de la dimensión especificada y los índices correspondientes.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Calcular el mínimo acumulativo
x = torch.tensor([3, 1, 4, 1, 5, 9, 2, 6])
values, indices = torch.cummin(x, dim=0)
print("Entrada:", x)
print("Mínimo acumulativo:", values)
print("Índices del mínimo:", indices)

# Tensor 2D
x = torch.tensor([[3, 1, 4], [1, 5, 9], [2, 6, 5]], dtype=torch.float32)
# Acumular por columnas
values_col, indices_col = torch.cummin(x, dim=0)
print("nAcumulado por columnas:")
print("Mínimo acumulativo:")
print(values_col)
print("Índices:")
print(indices_col)

# Acumular por filas
values_row, indices_row = torch.cummin(x, dim=1)
print("nAcumulado por filas:")
print("Mínimo acumulativo:")
print(values_row)

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

Otras extensiones