• Alykhan Tejani's avatar
    Add mode to ToPILImage constructor (#300) · 901c1ad2
    Alykhan Tejani authored
    * add assert for byte tensor to_pil_image
    
    * add mode param to ToPILImage
    
    * add some more tests
    
    * fix formatiing on error messages
    
    * added tests for to_pil_image with mode param
    
    * flake8 fixes
    901c1ad2
transforms.py 34.7 KB