Число осей в PyTorch

Ось - это направление, вдоль которого лежат элементы. Ряд чисел в одну строку имеет одну ось. Таблица из рядов и столбцов - две оси.

Число осей возвращает метод dim. Для ряда из нескольких чисел результат равен 1:

import torch

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

Длина ряда может быть любой, но число осей остаётся 1, пока данные не оформлены таблицей:

import torch

marks = torch.tensor([1, 2])
print(marks.dim()) # выведет 1

У таблицы метод dim вернёт 2:

import torch

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

Соберите ряд из чисел 3, 6 и 9 и выведите, сколько у него осей.

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

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