Method to

pointersect/inference/structures.py:1034–1039

(self, device: torch.device)

        self.directions_w = directions_w

    def to(self, device: torch.device) -> 'Ray':
        for attr_name in ['origins_w', 'directions_w']:
            arr = getattr(self, attr_name, None)
            if arr is not None:
                setattr(self, attr_name, arr.to(device=device))
        return self

    def reshape(self, *shape: T.List[int]):
        self.origins_w = self.origins_w.reshape(*shape)