I have a PyTorch model that produces a list of coordinates, e.g.
>>> coords = model(inputs) tensor([[1.1000, 3