Thanks for getting back. I tried to remove the the 3 from the target and replaced it with a 1 since it was in NumPy before passing it into a tensor and it worked . Would you consider it to be the right way to do it?