Новая ось в TensorFlow

Иной раз одномерный ряд нужно представить как строку или столбец таблицы. Ось длины один добавляет функция expand_dims: она вставляет новую ось в указанном месте, не меняя сами числа.

Возьмём ряд из трёх чисел и добавим ось перед элементами. Получится таблица из одной строки. Выведем результат и форму:

import tensorflow as tf

row = tf.constant([1, 2, 3])
as_row = tf.expand_dims(row, axis=0)
print(as_row)
print(as_row.shape)

Если новую ось поставить после элементов, ряд станет столбцом. Выведем такой вариант и его форму:

import tensorflow as tf

row = tf.constant([1, 2, 3])
as_col = tf.expand_dims(row, axis=1)
print(as_col)
print(as_col.shape)

Исходный ряд остаётся одномерным. Сравним формы ряда и строки таблицы:

import tensorflow as tf

row = tf.constant([4, 5, 6, 7])
line = tf.expand_dims(row, axis=0)
print(row.shape)
print(line.shape)

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

Соберите тензор из чисел 1, 2 и 3. Сделайте из него столбец таблицы и выведите форму результата.

Соберите тензор из пяти нулей. Один раз представьте его строкой таблицы, другой раз столбцом. Выведите обе формы.