YouTip

# Calculate Cumulative Minimum x = torch.tensor([3,1,4,1,5,9,2,6]) values, indices = torch.cummin(x, dim=0) print("Input:", x) print("Cumulative minimum:", values) print("Minimum Value Index:", indices) # 2 n-dimensional tensor x = torch.tensor([[3,1,4],[1,5,9],[2,6,5]], dtype=torch.float32) # Cumulative along columns values_col, indices_col = torch.cummin(x, dim=0) print("nCumulative along columns:") print("Cumulative minimum:") print(values_col) print("Index:") print(indices_col) # Cumulative along rows values_row, indices_row = torch.cummin(x, dim=1) print("nCumulative along rows:") print("Cumulative minimum:") print(values_row)

---

Pytorch torch Reference