(self, indices: torch.Tensor)
| 72 | return diffs.abs().argmin(dim=-1) |
| 73 | |
| 74 | def dequantize(self, indices: torch.Tensor) -> torch.Tensor: |
| 75 | return self.centroids.to(indices.device)[indices.long()] |
| 76 | |
| 77 | def __repr__(self) -> str: |
| 78 | return ( |
no outgoing calls