torch.round

torch.round(input, *, decimals=0, out=None) → Tensor

Округляет элементы input до ближайшего целого числа.

Для целочисленных входных данных, следуя соглашению array-api, возвращается копия входного тензора. Тип возвращаемого значения совпадает с типом данных входного значения.

Примечание

Эта функция использует правило «округления половин к чётному» для разрешения ситуаций, когда число равноудалено от двух целых чисел (например, round(2.5) равно 2).

Если указан аргумент :attr:`decimals`, используется алгоритм, аналогичный around NumPy. Этот алгоритм быстр, но неточен и может легко приводить к переполнению при использовании типов данных с низкой точностью. Например, round(tensor([10000], dtype=torch.float16), decimals=3) — это inf.

См. также

torch.ceil(), которая округляет вверх. torch.floor(), которая округляет вниз. torch.trunc(), которая округляет к нулю.

Параметры:

• input (Tensor) – входной тензор.
• decimals (int) – количество знаков после запятой для округления (по умолчанию: 0). Если значение decimals отрицательное, оно задаёт количество разрядов слева от десятичной точки.

Именованные аргументы:

out (Tensor, необязательный) – выходной тензор.

Пример:

>>> torch.round(torch.tensor((4.7, -2.3, 9.1, -7.7)))
tensor([ 5.,  -2.,  9., -8.])

>>> # Values equidistant from two integers are rounded towards the
>>> #   nearest even value (zero is treated as even)
>>> torch.round(torch.tensor([-0.5, 0.5, 1.5, 2.5]))
tensor([-0., 0., 2., 2.])

>>> # A positive decimals argument rounds to that decimal place
>>> torch.round(torch.tensor([0.1234567]), decimals=3)
tensor([0.1230])

>>> # A negative decimals argument rounds to the left of the decimal
>>> torch.round(torch.tensor([1200.1234567]), decimals=-3)
tensor([1000.])