Update

[ghstack-poisoned]
pytorch · Apr 29, 2024 · 5cad1e7 · 5cad1e7
1 parent c69c690
commit 5cad1e7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torch/_refs/__init__.py b/torch/_refs/__init__.py
@@ -3715,7 +3715,10 @@ def _reshape_view_helper(a: TensorLikeType, *shape, allow_copy: bool) -> TensorL
 
     # Squeezes tail
     while idx < a_.ndim:
-        assert a_.shape[idx] == 1
+        torch._check(
+            a_.shape[idx] == 1,
+            lambda: f"a.size({idx}) expected to be 1 but got {a_.shape[idx]}",
+        )
         a_ = squeeze(a_, idx)
 
     return a_