213d64f52eb9f33f6d127e5033e529c6694d3c89,torchgeometry/homography_warper.py,,create_meshgrid,#Any#Any#Any#,14

Before Change

    else:
        xs = torch.linspace(0, width - 1, width)
        ys = torch.linspace(0, height - 1, height)
    return torch.stack(torch.meshgrid([ys, xs])).view(1, 2, -1)[:, (1, 0), :]


// layer api

After Change

    def warp_grid(self, H):
        
        :param H: Homography or homographies (stacked) to transform all points
                  in the grid.
        :returns: Tensor[1, Height, Width, 2] containing transformed points in
                  normalized images space.
        
        batch_size = H.shape[0]  // expand grid to match the input batch size

Instances