Method
__init__
(
self,
origins_w: torch.Tensor, # (b, *m_shape, 3)
directions_w: torch.Tensor, # (b, *m_shape, 3)
)
Source from the content-addressed store, hash-verified
| 1024 | |
| 1025 | class Ray: |
| 1026 | def __init__( |
| 1027 | self, |
| 1028 | origins_w: torch.Tensor, # (b, *m_shape, 3) |
| 1029 | directions_w: torch.Tensor, # (b, *m_shape, 3) |
| 1030 | ): |
| 1031 | self.origins_w = origins_w |
| 1032 | self.directions_w = directions_w |
| 1033 | |
| 1034 | def to(self, device: torch.device) -> 'Ray': |
| 1035 | for attr_name in ['origins_w', 'directions_w']: |
Tested by
no test coverage detected