Hey Guys,
I want to normalize my Tensor.
When calling Normalize((0.5,0.5,0.5), (0.5,0.5,0.5)), I’m getting following error:

tensor.sub_(mean[:, None, None]).div_(std[:, None, None])
RuntimeError: expected device cpu but got device cuda:0

How do I send normalize to a cuda device? Is this even possible?
Thanks for helping.