On this page
torch.cummax
torch.cummax(input, dim, *, out=None)
-
Returns a namedtuple
(values, indices)
wherevalues
is the cumulative maximum of elements ofinput
in the dimensiondim
. Andindices
is the index location of each maximum value found in the dimensiondim
.- Parameters
- Keyword Arguments
-
out (tuple, optional) – the result tuple of two output tensors (values, indices)
Example:
>>> a = torch.randn(10) >>> a tensor([-0.3449, -1.5447, 0.0685, -1.5104, -1.1706, 0.2259, 1.4696, -1.3284, 1.9946, -0.8209]) >>> torch.cummax(a, dim=0) torch.return_types.cummax( values=tensor([-0.3449, -0.3449, 0.0685, 0.0685, 0.0685, 0.2259, 1.4696, 1.4696, 1.9946, 1.9946]), indices=tensor([0, 0, 2, 2, 2, 5, 6, 6, 8, 8]))
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.cummax.html