torch.amin
-
torch.amin(input, dim, keepdim=False, *, out=None) → Tensor
-
Returns the minimum value of each slice of the
input
tensor in the given dimension(s)dim
.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, whilemax(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 asinput
except in the dimension(s)dim
where they are of size 1. Otherwise,dim`s are squeezed (see :func:`torch.squeeze
), resulting in the output tensors having fewer dimensions thaninput
.- Parameters
- Keyword Arguments
-
out (Tensor, optional) – the output tensor.
Example:
>>> a = torch.randn(4, 4) >>> a tensor([[ 0.6451, -0.4866, 0.2987, -1.3312], [-0.5744, 1.2980, 1.8397, -0.2713], [ 0.9128, 0.9214, -1.7268, -0.2995], [ 0.9023, 0.4853, 0.9075, -1.6165]]) >>> torch.amin(a, 1) tensor([-1.3312, -0.5744, -1.7268, -1.6165])
-
© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.amin.html