Argmin
self | (Tensor) the input tensor. |
---|---|
dim | (int) the dimension to reduce. If |
keepdim | (bool) whether the output tensor has |
Returns the indices of the minimum value of all elements in the input
tensor.
This is the second value returned by torch_min
. See its
documentation for the exact semantics of this method.
Returns the indices of the minimum values of a tensor across a dimension.
This is the second value returned by torch_min
. See its
documentation for the exact semantics of this method.
if (torch_is_installed()) { a = torch_randn(c(4, 4)) a torch_argmin(a) a = torch_randn(c(4, 4)) a torch_argmin(a, dim=1) }#> torch_tensor #> 1 #> 2 #> 4 #> 4 #> [ CPULongType{4} ]