Ячейка таблицы в PyTorch

Для одной ячейки таблицы в квадратных скобках через запятую пишут номер строки и номер столбца. Оба номера считаются с нуля. Запись [i, j] задаёт одну ячейку сразу, а не две скобки подряд.

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

import torch

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

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

import torch

scores = torch.tensor([
	[10, 20],
	[30, 40],
	[50, 60],
])
print(scores[2, 0]) # выведет tensor(50)
print(scores[1, 1]) # выведет tensor(40)

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

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

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