torch_amax function

Amax

torch_amax(self, dim = list(), keepdim = FALSE)

Arguments

• self: (Tensor) the input tensor.
• dim: (int or tuple of ints) the dimension or dimensions to reduce.
• keepdim: (bool) whether the output tensor has dim retained or not.

Note

The difference between max/min and amax/amin is:

• amax/amin supports reducing on multiple dimensions,
• amax/amin does not return indices,
• amax/amin evenly distributes gradient between equal values, while max(dim)/min(dim) propagates gradient only to a single index in the source tensor.

If keepdim isTRUE, the output tensors are of the same size asinputexcept in the dimension(s)dimwhere they are of size 1. Otherwise,dims are squeezed (see [torch_squeeze()]), resulting in the output tensorshaving fewer dimension thaninput`.

amax(input, dim, keepdim=FALSE, *, out=None) -> Tensor

Returns the maximum value of each slice of the input tensor in the given dimension(s) dim.

Examples

if (torch_is_installed()) {

a <- torch_randn(c(4, 4))
a
torch_amax(a, 1)
}