I tried this operation for one batch and worked.
x = torch.FloatTensor([[ax,bx],[cx,dx],[ex,fx],[gx,hx]]) y = torch.FloatTensor([[ay,by],[cy,dy],[ey,fy],[gy,h