pytorch：如何堆叠 2 个张量

【问题标题】：pytorch : How to stack 2 tensorspytorch：如何堆叠 2 个张量
【发布时间】：2020-12-23 13:50:31
【问题描述】：

我正在尝试将 2 个张量 A.shape=(64,16,16) 和 B.shape=(64,16,16) 堆叠在一个形状为 C.shape=(1,128,16,16) 的张量中

我尝试过的函数没有在哪里工作
torch.stack => C.shape=(2,64,16,16) 和 torch.cat => C.shape=(128,16,16)

谁能帮帮我

【问题讨论】：

【解决方案1】：

先Concat，然后使用unsqueeze在0th位置添加单例维度

torch.cat([A, B]).unsqueeze(0)

【讨论】：