Pytorch torch.stack torch.cat

torch.stack和torch.cat

torch.stack实对张量沿指定维度进行级联

torch.stack和torch.cat的区别,前一个会增加维度,而后一个不变

你可能感兴趣的:(Pytorch)