下面这种是错误的,梯度会消失
data = torch.tensor([[0,0],[x1,y1],[x2,y2],[1,1]],requires_grad=True).cuda()
用torch.cat,其他方法好像不行。