(self, device: torch.device)
| 2186 | return rgbd_images |
| 2187 | |
| 2188 | def to(self, device: torch.device) -> 'RGBDImage': |
| 2189 | for attr_name in ['rgb', 'depth', 'normal_w', 'hit_map', 'camera', 'feature']: |
| 2190 | arr = getattr(self, attr_name, None) |
| 2191 | if arr is not None: |
| 2192 | setattr(self, attr_name, arr.to(device=device)) |
| 2193 | return self |
| 2194 | |
| 2195 | def detach(self) -> 'RGBDImage': |
| 2196 | for attr_name in ['rgb', 'depth', 'normal_w', 'hit_map', 'camera', 'feature']: |