将一个（768，1）的张量转换为（1024，）

我想要将一个（768，1）的张量转换为（1024，）
试了下面几个方法都没有得到目标结果

（变成了768，）：

outputs = outputs.flatten()[:1024]  # 将二维数组压缩为一维，如果维度超过1024，则截断；如果不足1024，则补0
numpy_outputs = outputs.squeeze().detach().cpu().numpy()[:1024]  # 将二维数组压缩为一维，如果维度超过1024，则截断；如果不足1024，则补0

变成了（1，）

numpy_outputs = np.mean(outputs.detach().cpu().numpy(), axis=1)[:1024]  # 将二维数组压缩为一维，如果维度超过1024，则截断；如果不足1024，则补0

最后通过零填充实现的

outputs = torch.cat([outputs.squeeze(), torch.zeros(1024 - outputs .shape[1]).cuda()])