Форма тензора в PyTorch

Форма показывает, сколько элементов приходится на каждую ось. У ряда из трёх чисел это одна величина - длина ряда.

Прочитать форму можно через атрибут shape. Результат имеет тип torch.Size и ведёт себя как кортеж размеров. Создадим ряд оценок и выведем его форму:

import torch

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

Если элементов больше, меняется только число в описании формы:

import torch

temps = torch.tensor([1, 2, 3, 4, 5])
print(temps.shape) # выведет torch.Size([5])

У таблицы форма состоит из двух чисел: число рядов и число столбцов в каждом ряду:

import torch

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

Соберите ряд из 2, 4, 6 и 8 и выведите описание формы этого ряда.

Соберите таблицу из рядов [1, 0] и [0, 1] и выведите её форму.

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