Растягивание формы в TensorFlow

Чтобы растянуть тензор до заданной формы, используют функцию broadcast_to. Она повторяет данные так, чтобы получилась нужная форма. Результат ведёт себя как представление: запись в такой тензор может быть недоступна, поэтому присваивать в него не стоит.

Ряд из четырёх чисел можно развернуть в таблицу (3, 4): каждая строка повторяет исходный ряд. Выведем растянутый тензор:

import tensorflow as tf

row = tf.constant([1, 2, 3, 4])
table = tf.broadcast_to(row, [3, 4])
print(table)
# выведет tf.Tensor([[1 2 3 4] [1 2 3 4] [1 2 3 4]], shape=(3, 4), dtype=int32)

Растяните ряд [5, 6] до формы (4, 2) и выведите результат.

Растяните ряд [0, 1, 2] до формы (2, 3) и выведите форму результата.

Растяните ряд [7] до формы (3, 1) и выведите получившуюся таблицу.