pytorch：如何堆疊 2 個張量

Question

我試圖堆疊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)

可以幫助我嗎？

Answer 1

先連接，然后使用unsqueeze在0th位置添加單例維度

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