Строка таблицы в TensorFlow

Если у двумерного тензора в квадратных скобках указать один индекс, вернётся целая строка как одномерный тензор. Номер строки считается с нуля сверху вниз.

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

import tensorflow as tf

grid = tf.constant([
	[1, 2],
	[3, 4],
	[5, 6],
])
print(grid[0])
print(grid[2])

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

import tensorflow as tf

scores = tf.constant([
	[10, 20, 30],
	[40, 50, 60],
	[70, 80, 90],
])
middle = scores[1]
print(middle)

Создайте таблицу из четырёх строк и трёх столбцов и выведите её верхнюю строку как отдельный ряд.

Соберите таблицу 2 на 4 из единиц и выведите нижнюю строку этой таблицы.

Создайте таблицу из пяти строк и двух столбцов с любыми числами и выведите третью строку сверху.