Получение формы (4, 1, 84, 84) с помощью пирога

Question

Предположим, у меня есть четыре тензора пикторов (tensor1, tensor2, tensor3, tensor4). Каждый тензор имеет форму (1, 1, 84, 84). Первое измерение - это число тензоров, второе измерение - это количество цветов (например, в нашем примере это оттенки серого), а последние два измерения представляют высоту и ширину изображения.

Я хочу сложить их так, чтобы получить форму (4, 1, 84, 84).

Я пытался torch.stack((tensor1, tensor2, tensor3, tensor4), dim=0), но я получил форму (4, 1, 1, 84, 84).

Как я могу сложить эти тензоры так, чтобы форма была (4, 1, 84, 84)

Answer 1

Вы можете использовать функцию сцепления:

a = torch.ones(1,1,84,84)
b = torch.ones(1,1,84,84)
c = torch.cat((a,b), 0) # size[2,1,84,84]