(self)
| 697 | ) |
| 698 | |
| 699 | def realize_valid_mask(self): |
| 700 | if self.valid_mask is not None: |
| 701 | return |
| 702 | b, n, _dim = self.xyz_w.shape |
| 703 | self.valid_mask = torch.ones( |
| 704 | b, n, 1, |
| 705 | dtype=torch.bool, device=self.xyz_w.device) |
| 706 | if self.included_point_at_inf: |
| 707 | self.valid_mask[:, 0] = 0 |
| 708 | |
| 709 | def reset_valid_mask(self): |
| 710 | """set all points as valid""" |
no outgoing calls
no test coverage detected