e,                  input.coords_key, None, input.coords_man)))          device = torch.device('cuda')         with torch.cuda.device(0):             input = input.to(device)             pool = pool.to(device)             output = pool(input)             print(output)      