6fac413c0c51c3be54de3a68801b43446ab98dee,graph/grid.py,,grid_points,#Any#,42

Before Change

    h, w = size
    x, y = np.meshgrid(np.arange(w), np.arange(h))
    x = torch.FloatTensor(x)
    y = torch.FloatTensor(y)
    points = torch.FloatTensor(h * w, 2)

    if torch.cuda.is_available():
        x = x.cuda()
        y = y.cuda()
        points = points.cuda()

    points[:, 0] = y.view(-1)
    points[:, 1] = x.view(-1)
    return points

After Change

        y = y.cuda()

    x = x.repeat(h)
    y = y.view(-1, 1).repeat(1, w).view(-1)

    return torch.cat((y, x)).view(2, -1).t()


size = torch.Size([3, 2])

Instances