torch.stack, torch.cat, torch.stack.max/mean/sum维度变换详解