423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-21 14:50:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

devices for transform3d

Browse Source 
Summary: Make `to` on Transform3D carry its member _transforms.

Reviewed By: nikhilaravi

Differential Revision: D25978611

fbshipit-source-id: 12b39e7a657f28d59ca60800bf9f4193a2c08197
This commit is contained in:
Jeremy Reizenstein
2021-01-21 04:56:56 -08:00
committed by Facebook GitHub Bot
parent 4711665edb
commit d60c52df4a
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pytorch3d/transforms/transform3d.py
View File

@@ -421,8 +421,9 @@ class Transform3d:
if self.device != device:
other.device = device
other._matrix = self._matrix.to(device=device, dtype=dtype)
for t in other._transforms:
t.to(device, copy=copy, dtype=dtype)
other._transforms = [
t.to(device, copy=copy, dtype=dtype) for t in other._transforms
]
return other
def cpu(self):