torch_round {torch}

Round

Description

Round

Usage

torch_round(self, decimals)

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)
}