我正在尝试堆叠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)
可以帮助我吗
Concat首先，然后用于unsqueeze在0th位置添加单身尺寸
torch.cat([A, B]).unsqueeze(0)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句