Removed expand_as() call in aminmax_backward

vijayabhaskar-ev · vijayabhaskar-ev · commit b7818f68a8c1 · 2025-07-21T20:49:47.000+05:30
diff --git a/torch/csrc/autograd/FunctionsManual.cpp b/torch/csrc/autograd/FunctionsManual.cpp
@@ -238,15 +238,13 @@ Tensor aminmax_backward(
 
   if (grad_min.defined()) {
     auto grad_min_full =
-        restore_reduced_dims(grad_min, dims, keepdim)
-            .expand_as(min_mask);          
+        restore_reduced_dims(grad_min, dims, keepdim);      
     result = scale_grad_by_count(grad_min_full, min_mask, dims);
   }
 
   if (grad_max.defined()) {
     auto grad_max_full =
-        restore_reduced_dims(grad_max, dims, keepdim)
-            .expand_as(max_mask);          
+        restore_reduced_dims(grad_max, dims, keepdim);          
     auto grad_max_res = scale_grad_by_count(grad_max_full, max_mask, dims);
     result = result.defined() ? result + grad_max_res
                               : grad_max_res;
