Spaces:
Runtime error
Runtime error
File size: 334 Bytes
5e88f62 |
1 2 3 4 5 6 7 8 9 10 |
import torch
def get_meshgrid(resolution, device):
grid_x, grid_y = torch.meshgrid(torch.arange(resolution[0]).float() / resolution[0],
torch.arange(resolution[1]).float() / resolution[1], indexing='ij')
grid_x = grid_x.to(device)
grid_y = grid_y.to(device)
return grid_x, grid_y
|