(self, device=torch.device("cpu"))
| 32 | return self._encode[device] |
| 33 | |
| 34 | def decode_lut(self, device=torch.device("cpu")): |
| 35 | if device not in self._decode: |
| 36 | cpu = torch.device("cpu") |
| 37 | self._decode[device] = tuple(e.to(device) for e in self._decode[cpu]) |
| 38 | return self._decode[device] |
| 39 | |
| 40 | def xyz2key(self, x, y, z, depth): |
| 41 | key = torch.zeros_like(x) |