modified the ToTensor transform to properly work with grasyscale images

Merge request reports

Loading