Добавление оси в PyTorch

Если тензору не хватает отдельной оси, её вставляют методом unsqueeze. Новая ось всегда имеет длину 1; существующие элементы не меняются и не дублируются.

У обычного ряда из трёх чисел одна ось. Добавим перед ним ось с индексом 0 и посмотрим на форму:

import torch

marks = torch.tensor([1, 2, 3])
batch = marks.unsqueeze(0)
print(batch.shape) # выведет torch.Size([1, 3])

Тот же ряд можно оформить как столбец: новая ось вставляется с индексом 1:

import torch

marks = torch.tensor([1, 2, 3])
column = marks.unsqueeze(1)
print(column.shape) # выведет torch.Size([3, 1])

Соберите ряд из чисел 4, 5 и 6 и выведите форму тензора после добавления оси перед этим рядом.

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

Соберите таблицу из одного ряда [2, 3] и выведите форму после вставки новой оси в начало.