Python pytorch：如何叠加2个张量
我试图将两个张量Python pytorch：如何叠加2个张量,python,stack,pytorch,concatenation,tensor,Python,Stack,Pytorch,Concatenation,Tensor,我试图将两个张量A.shape=（64,16,16）和B.shape=（64,16,16）叠加在一个shapeC.shape=（1128,16,16） 我尝试过的函数中没有一个是在 torch.stack=>C.shape=（2,64,16,16）和 torch.cat=>C.shape=（128,16,16） enyones能否帮助我先进行Concat，然后使用在第0个位置添加singleton维度 torch.cat([A, B]).unsqueeze(0) 先连接，然后使用在0th位置添
A.shape=（64,16,16）B.shape=（64,16,16）C.shape=（1128,16,16）torch.stackC.shape=（2,64,16,16）torch.catC.shape=（128,16,16）torch.cat([A, B]).unsqueeze(0)
0thtorch.cat([A, B]).unsqueeze(0)