PyTorch torch.cummax Function
PyTorch torch Reference Manual
torch.cummaxIt is a function in PyTorch used to compute cumulative maxima. It returns the cumulative maximum along a specified dimension along with the corresponding indices.
Function Definition
torch.cummax(input, dim, dtype=None)
Usage Example
Example
import torch
# Compute cumulative maximum
x = torch.tensor([3, 1, 4, 1, 5, 9, 2, 6])
values, indices = torch.cummax(x, dim=0)
print("Input:", x)
print("Cumulative maximum:", values)
print("Maximum indices:", indices)
# 2D tensor
x = torch.tensor([[3, 1, 4], [1, 5, 9], [2, 6, 5]], dtype=torch.float32)
# Accumulate by column
values_col, indices_col = torch.cummax(x, dim=0)
print("nAccumulate by column:")
print("Cumulative maximum:")
print(values_col)
print("Indices:")
print(indices_col)
# Accumulate by row
values_row, indices_row = torch.cummax(x, dim=1)
print("nAccumulate by row:")
print("Cumulative maximum:")
print(values_row)
# Compute cumulative maximum
x = torch.tensor([3, 1, 4, 1, 5, 9, 2, 6])
values, indices = torch.cummax(x, dim=0)
print("Input:", x)
print("Cumulative maximum:", values)
print("Maximum indices:", indices)
# 2D tensor
x = torch.tensor([[3, 1, 4], [1, 5, 9], [2, 6, 5]], dtype=torch.float32)
# Accumulate by column
values_col, indices_col = torch.cummax(x, dim=0)
print("nAccumulate by column:")
print("Cumulative maximum:")
print(values_col)
print("Indices:")
print(indices_col)
# Accumulate by row
values_row, indices_row = torch.cummax(x, dim=1)
print("nAccumulate by row:")
print("Cumulative maximum:")
print(values_row)
Other Extensions