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

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

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

import torch

grid = torch.tensor([
	[1, 2],
	[3, 4],
	[5, 6],
])
print(grid[0]) # выведет tensor([1, 2])
print(grid[2]) # выведет tensor([5, 6])

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

import torch

scores = torch.tensor([
	[10, 20, 30],
	[40, 50, 60],
	[70, 80, 90],
])
middle = scores[1]
print(middle) # выведет tensor([40, 50, 60])
print(middle.ndim) # выведет 1

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

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

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