Сумма по оси в TensorFlow

У функции reduce_sum есть параметр axis: тогда сводка идёт не по всем ячейкам, а вдоль выбранной линии. Ось с номером 0 идёт по столбцам, сверху вниз. Ось с номером 1 идёт по строкам, слева направо.

Возьмём таблицу из двух строк и трёх столбцов. Сложим по столбцам и выведем результат вместе с его формой:

import tensorflow as tf

table = tf.constant([[1, 2, 3], [4, 5, 6]])
by_cols = tf.reduce_sum(table, axis=0)
print(by_cols)
print(by_cols.shape)

Теперь сложим по строкам и снова посмотрим форму:

import tensorflow as tf

table = tf.constant([[1, 2, 3], [4, 5, 6]])
by_rows = tf.reduce_sum(table, axis=1)
print(by_rows)
print(by_rows.shape)

Для таблицы цен [[10, 20], [30, 40]] сложите по столбцам и выведите ряд сумм и его форму.

Для таблицы баллов [[1, 2, 3], [4, 5, 6]] сложите по строкам и выведите ряд сумм и его форму.

Для таблицы отметок [[2, 0], [1, 3], [4, 1]] сложите по столбцам и выведите ряд сумм.