• Roman Shapovalov's avatar
    Adding utility methods to TensorProperties · 0c02ae90
    Roman Shapovalov authored
    Summary:
    Context: in the code we are releasing with CO3D dataset, we use  `cuda()` on TensorProperties like Pointclouds and Cameras where we recursively move batch to a GPU. It would be good to push it to a release so we don’t need to depend on the nightly build.
    
    Additionally, I aligned the logic of `.to("cuda")` without device index to the one of `torch.Tensor` where the current device is populated to index. It should not affect any actual use cases but some tests had to be changed.
    
    Reviewed By: bottler
    
    Differential Revision: D29659529
    
    fbshipit-source-id: abe58aeaca14bacc68da3e6cf5ae07df3353e3ce
    0c02ae90
test_transforms.py 43.5 KB