err on dot product for tensors of different sizes (#106572)

bobby-palmer · pytorchmergebot · commit 3e6da46aff84 · 2023-08-04T18:34:34.000Z
Fixes #106448 Pull Request resolved: #106572 Approved by: https://github.com/ezyang
diff --git a/torch/_meta_registrations.py b/torch/_meta_registrations.py
@@ -1696,6 +1696,14 @@ def dot_check(self, other):
         lambda: f"1D tensors expected, but got {self.dim()}D and {other.dim()}D tensors",
     )
 
+    def numel_error():
+        return (
+            f"inconsistent tensor size, expected tensor [{self.numel()}] and src [{other.numel()}] to have the"
+            f"same number of elements, but got {self.numel()} and {other.numel()} elements respectively"
+        )
+
+    torch._check(self.numel() == other.numel(), numel_error)
+
 
 @register_meta(aten.dot.default)
 def meta_dot(self, tensor):
