• Brian Vaughan's avatar
    make convert_image_dtype scriptable (#2485) · c542137c
    Brian Vaughan authored
    
    
    * make convert_image_dtype scriptable
    
    * move convert dtype to functional_tensor since only works on tensors
    
    * retain availability of convert_image_dtype in functional.py
    
    * Update code and tests
    
    * Replaced int by torch.dtype
    
    * int -> torch.dtype and use F instead of F_t
    
    * Update functional_tensor.py
    Co-authored-by: default avatarvfdev-5 <vfdev.5@gmail.com>
    c542137c
test_functional_tensor.py 32.9 KB