diff --git a/tensorizer/serialization.py b/tensorizer/serialization.py index 3b14e7ea..af4c3716 100644 --- a/tensorizer/serialization.py +++ b/tensorizer/serialization.py @@ -221,7 +221,7 @@ def deserialized_length(self): if self.data_length > 0: return self.data_length element_size: int = numpy.dtype(self.dtype).itemsize - num_elements: int = numpy.product(self.shape) + num_elements: int = numpy.prod(self.shape) return element_size * num_elements