torch.shape()三维矩阵降为二维