(self)
| 528 | description: str |
| 529 | |
| 530 | def load(self) -> Tensor: |
| 531 | ret = self._load() |
| 532 | # Should be okay if it maps to the same numpy type? |
| 533 | assert ret.data_type == self.data_type or (self.data_type.dtype == ret.data_type.dtype), \ |
| 534 | (self.data_type, ret.data_type, self.description) |
| 535 | return ret |
| 536 | |
| 537 | def astype(self, data_type: DataType) -> LazyTensor: |
| 538 | self.validate_conversion_to(data_type) |