Среднее в TensorFlow

Среднее арифметическое по всем ячейкам даёт функция reduce_mean. Она принимает тензор и возвращает одно число, если сводка идёт по всем элементам сразу.

Создадим ряд из дробных чисел и выведем среднее:

import tensorflow as tf

scores = tf.constant([4.0, 5.0, 3.0])
aver = tf.reduce_mean(scores)
print(aver)

Целые из Python дают int32. Среднее по ним тоже получается целым и теряет дробную часть. Перед reduce_mean такие числа переводят в дробный тип. Создадим целочисленный ряд, приведём его к дробному типу и выведем среднее:

import tensorflow as tf

marks = tf.constant([1, 2, 2])
aver = tf.reduce_mean(tf.cast(marks, tf.float32))
print(aver)
# выведет tf.Tensor(1.6666666, shape=(), dtype=float32)

Соберите ряд цен 2, 4, 6 и 8 и выведите среднее арифметическое одним числом.

Соберите ряд баллов 1, 2 и 4 и выведите среднее одним числом.

Соберите таблицу отметок [[2, 4], [6, 8]] и выведите среднее по всем её ячейкам.