torchvision/transforms/functional_tensor.py · 11a39aaab5b55a3c116c2e8d8001bad94a96f99d · OpenDAS / vision

Make RandomHorizontalFlip torchscriptable (#2282) · 11a39aaa

Francisco Massa authored Jun 04, 2020

* Make RandomHorizontalFlip torchscriptable

* Make _is_tensor_a_torch_image more generic

* Make RandomVerticalFlip torchscriptable (#2283)

* Make RandomVerticalFlip torchscriptable

* Fix lint

11a39aaa

functional_tensor.py 7.9 KB

Replace functional_tensor.py