Функция round
Функция round применяется к тензору
и округляет каждый его элемент до ближайшего
целого значения. Первым параметром функция
принимает тензор с числами. Округление
выполняется по правилам банковского округления:
если дробная часть равна 0.5, результат
округляется к ближайшему четному числу.
Синтаксис
tf.round(x, [name])
Пример
Давайте округлим тензор с дробными числами:
import tensorflow as tf
t = tf.constant([1.4, 2.6, 3.5, 4.5, 5.1])
res = tf.round(t)
print(res)
Результат выполнения кода:
tf.Tensor([1. 3. 4. 4. 5.], shape=(5,), dtype=float32)
Пример
Давайте округлим двумерный тензор:
import tensorflow as tf
t = tf.constant([[1.2, 2.8, 3.5], [4.5, 5.3, 6.7]])
res = tf.round(t)
print(res)
Результат выполнения кода:
tf.Tensor(
[[1. 3. 4.]
[4. 5. 7.]], shape=(2, 3), dtype=float32)
Пример
Давайте округлим отрицательные числа и числа
с дробной частью 0.5:
import tensorflow as tf
t = tf.constant([-1.5, -2.5, 1.5, 2.5, -3.4])
res = tf.round(t)
print(res)
Результат выполнения кода:
tf.Tensor([-2. -2. 2. 2. -3.], shape=(5,), dtype=float32)
Пример
Давайте преобразуем округленный тензор в массив NumPy
с помощью метода numpy:
import tensorflow as tf
t = tf.constant([1.4, 2.6, 3.5, 4.5, 5.1])
res = tf.round(t)
print(res.numpy())
Результат выполнения кода:
[1. 3. 4. 4. 5.]
