changed to view

CTUAvastLab · Apr 4, 2024 · 4268465 · 4268465
1 parent 7dc4b21
commit 4268465
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/aggregations/segmented_max.jl b/src/aggregations/segmented_max.jl
@@ -67,9 +67,10 @@ function segmented_max_back(Δ, y, x, ψ, bags)
             end
         else
             fi = first(b)
-            for k in axes(v,1)
-                v[k] = x[k,fi]
-            end
+            v .= @view x[:,fi]
+            # for k in axes(v,1)
+            #     v[k] = x[k,fi]
+            # end
             idxs .= fi
             for j in b
                 for i in axes(x,1)