Hi,



 ben_mi:

values = torch.arange(23, dtype=torch.uint8, device="cuda:0")


This line generates a list [0, 1, ..., 22]. If you want just a single number, close the range on it [23, 24].
values = torch.arange(23, 24, dtype=torch.uint8, device="cuda:0")

Bests