Понимание Тензорной Формы PyTorch

Question

У меня есть простой вопрос относительно формы тензора, который мы определяем в PyTorch.Скажем, если я скажу:

input = torch.randn(32, 35)

Это создаст матрицу с 32 строками и 35 столбцами.Теперь, когда я определяю:

input2 = torch.randn(1,2,32, 35)

Что я могу сказать о размерности новой матрицы input2?Как я могу определить строки и столбцы здесь?Я имею в виду, у меня есть две матрицы с формами 32 * 35, упакованные тензором?

Я хочу лучше понять геометрию этого.Благодарю.

Answer 1

Да, это правильно.Ваш тензор input2 имеет ранг 4. (Ранг - это измерение), а границы каждого измерения - (1,2,32,35)

1. Первое измерение может содержать один элемент.
2. Второй может содержать два элемента.
3. Третий может содержать 32 элемента.
4. Четвертое измерение может содержать 35 элементов.

РЕДАКТИРОВАТЬ: Я считаю, что полезно думать о многомерных массивах как о серии списков.В вашем случае, тензор ранга 4, будет списком списков списков списков.