From 938bc4d5036a26631798d3611bd5f2907bfb64aa Mon Sep 17 00:00:00 2001
From: Dmitry Kurtaev <dmitry.kurtaev@gmail.com>
Date: Fri, 22 Dec 2023 15:45:18 +0300
Subject: [PATCH] [CUDA] Hotfix Scale with 1 parameter

---
 modules/dnn/src/cuda4dnn/primitives/scale_shift.hpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules/dnn/src/cuda4dnn/primitives/scale_shift.hpp b/modules/dnn/src/cuda4dnn/primitives/scale_shift.hpp
index 9da7ec3326..abe2615590 100644
--- a/modules/dnn/src/cuda4dnn/primitives/scale_shift.hpp
+++ b/modules/dnn/src/cuda4dnn/primitives/scale_shift.hpp
@@ -128,6 +128,9 @@ namespace cv { namespace dnn { namespace cuda4dnn {
 
             /* the scale shift operation might require broadcasting */
             const int end_axis = [&] {
+                if (num_parameters == 1) {
+                    return static_cast<int>(axis + 1);
+                }
                 for (int endAxis = axis + 1; endAxis <= input.rank(); endAxis++) {
                     if (input.size_range(axis, endAxis) == mid_size)
                         return endAxis;