Транспонирование в TensorFlow

У табличного тензора строки и столбцы меняет функция transpose. Для двумерной таблицы первая строка становится первым столбцом и так далее.

Создадим таблицу из двух строк и трёх столбцов. Выведем форму до преобразования, саму таблицу после него и новую форму:

import tensorflow as tf

table = tf.constant([[1, 2, 3], [4, 5, 6]])
print(table.shape)
turned = tf.transpose(table)
print(turned)
print(turned.shape)

Число элементов не меняется: меняются только длины осей. У таблицы 2×3 после преобразования форма становится 3×2:

import tensorflow as tf

grid = tf.constant([[10, 20], [30, 40], [50, 60]])
print(grid.shape)
turned = tf.transpose(grid)
print(turned.shape)

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

Соберите таблицу из одной строки [7, 8, 9, 10]. Поменяйте строки и столбцы местами и выведите форму результата.

Соберите табличный тензор формы (2, 4) из нулей. Выведите форму исходного тензора и форму после обмена строк и столбцов.