Функция rint
Функция rint округляет каждый элемент
тензора до ближайшего целого числа. Если
дробная часть числа равна 0.5, то
округление происходит до ближайшего четного
целого числа. Первым параметром функция
принимает тензор или число, которое нужно
округлить.
Синтаксис
tf.rint(x, [name])
Пример
Давайте округлим тензор с положительными и отрицательными дробными числами:
import tensorflow as tf
t = tf.constant([1.2, 2.5, 3.7, -1.2, -2.5, -3.7])
res = tf.rint(t)
print(res)
Результат выполнения кода:
tf.Tensor([ 1. 2. 4. -1. -2. -4.], shape=(6,), dtype=float32)
Пример
Давайте округлим тензор с четными и
нечетными значениями 0.5:
import tensorflow as tf
t = tf.constant([0.5, 1.5, 2.5, 3.5])
res = tf.rint(t)
print(res)
Результат выполнения кода:
tf.Tensor([0. 2. 2. 4.], shape=(4,), dtype=float32)
Пример
Давайте применим функцию rint
к двумерному тензору:
<+python+>
import tensorflow as tf
t = tf.constant([[1.1, 2.5, 3.9], [4.2, 5.5, 6.8]])
res = tf.rint(t)
print(res)
<-python+>
Результат выполнения кода:
tf.Tensor(
[[1. 2. 4.]
[4. 6. 7.]], shape=(2, 3), dtype=float32)