math

Author: justinchuby

onnxscript/ir/_core.py

@@ -114,7 +114,7 @@ def _repr_base(self) -> str:
     @property
     def size(self) -> int:
         """The number of elements in the tensor."""
-        return np.prod(self.shape.numpy())  # type: ignore[return-value,attr-defined]
+        return math.prod(self.shape.numpy())  # type: ignore[attr-defined]
 
     @property
     def nbytes(self) -> int: