Round
Arguments
- self
(Tensor) the input tensor.
- decimals
Number of decimal places to round to (default: 0). If decimals is negative, it specifies the number of positions to the left of the decimal point.
round(input, out=NULL) -> Tensor
Returns a new tensor with each of the elements of input rounded
to the closest integer.
Examples
if (torch_is_installed()) {
a = torch_randn(c(4))
a
torch_round(a)
}
#> torch_tensor
#> 0
#> -1
#> -0
#> -1
#> [ CPUFloatType{4} ]