As the error message explains, Normalize expects an image tensor in the shape [channels, height, width] as its input, while you are trying to pass the target tensor to it, which won’t work.
If you want to normalize the target, you could manually subtract the mean and divide by the standard deviation. However, make sure you really want and need to normalize the target, as it’s often not wanted.