• Varun Agrawal's avatar
    Fix for #409 (#673) · 2d493d6a
    Varun Agrawal authored
    * added separate checks for dimensionality in to_pil_image and added tests
    
    * updated to_pil_image to use both 2D ndarrays and tensors, as well as refactored the tests
    2d493d6a
test_transforms.py 48.8 KB