Добавление 1D-тензоров итеративно для создания 2D-тензоров

Question

У меня есть тензор длиной 64, и я пытаюсь добавить к 1-му измерению рекурсивный 2D-тензор. По существу, при каждой итерации создается новый тензор длины 1D 64, и он повторяется 6 раз, поэтому к концу у меня должен быть тензор 6 x 64. Как мне это сделать?

Я пытался

names_tensor = torch.cat((names_tensor, sampled_indexes), dim=1)

Где name_tensor инициируется как torch.zeros(0), а sampled_indexes - это тензор длины 64, который добавляется при каждой итерации

Answer 1

import torch
names_tensor = torch.zeros(0)
sampled_indexes = torch.ones(64)

names_tensor = torch.cat((names_tensor, sampled_indexes.repeat(6))).reshape(6,64)
names_tensor.size()
>> torch.Size([6, 64])

Используйте torch.cat() для последовательности конкатенаций сначала тензоров, а затем измените тензор с помощью torch.reshape()