R/gen-namespace-docs.R
, R/gen-namespace-examples.R
, R/gen-namespace.R
torch_median.Rd
Median
torch_median(self, dim, keepdim = FALSE)
self | (Tensor) the input tensor. |
---|---|
dim | (int) the dimension to reduce. |
keepdim | (bool) whether the output tensor has |
Returns the median value of all elements in the input
tensor.
Returns a namedtuple (values, indices)
where values
is the median
value of each row of the input
tensor in the given dimension
dim
. And indices
is the index location of each median value found.
By default, dim
is the last dimension of the input
tensor.
If keepdim
is TRUE
, the output tensors are of the same size
as input
except in the dimension dim
where they are of size 1.
Otherwise, dim
is squeezed (see torch_squeeze
), resulting in
the outputs tensor having 1 fewer dimension than input
.
if (torch_is_installed()) { a = torch_randn(c(1, 3)) a torch_median(a) a = torch_randn(c(4, 5)) a torch_median(a, 1) }#> [[1]] #> torch_tensor #> 0.2246 #> -0.5522 #> -0.1973 #> -0.6693 #> -0.0877 #> [ CPUFloatType{5} ] #> #> [[2]] #> torch_tensor #> 1 #> 2 #> 3 #> 1 #> 2 #> [ CPULongType{5} ] #>