tensor类型数组一维变二维

def tran_predy(y_pred):
    a = y_pred.detach().numpy()#先将tensor转换成numpy数组
    c = []#一维数组变二维
    for i in a:
        tmp = [float(i)]
        c.append(tmp)
    d = torch.tensor(c)#变成tensor类型
    return d

输入 

tensor([4, 4, 4, 4, 4, 4, 4, 4, 4, 4])

输出 

tensor([[4.],
        [4.],
        [4.],
        [4.],
        [4.],
        [4.],
        [4.],
        [4.],
        [4.],
        [4.]])