Элемент одномерного тензора в PyTorch

К одной ячейке тензора-ряда обращаются через индекс в квадратных скобках. Первый элемент имеет индекс 0, второй 1, третий 2. После имени тензора в скобках пишут нужный номер.

Создадим ряд оценок и выведем элемент с индексом 0 и с индексом 2:

import torch

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

Индекс может быть любым целым от нуля до последней ячейки. Для ряда из четырёх чисел последний индекс равен 3:

import torch

temps = torch.tensor([10, 20, 30, 40])
print(temps[1]) # выведет tensor(20)
print(temps[3]) # выведет tensor(40)

Соберите ряд из чисел 7, 8 и 9 и выведите его первую ячейку.

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

Соберите ряд из пяти дробных чисел и выведите ячейку с номером 2, если первая ячейка имеет номер 0.