Add missing .get_final_data_gradient() for repeat layer

9f6aefc0 · Davis King · 1516cf31 · 9f6aefc0
Commit 9f6aefc0 authored Jan 28, 2021 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

dlib/dnn/core.h dlib/dnn/core.h +3 -0

No files found.
--- a/dlib/dnn/core.h
+++ b/dlib/dnn/core.h
@@ -1901,6 +1901,9 @@ namespace dlib
            return details[0].get_gradient_input();
        }

+        const tensor& get_final_data_gradient(
+        ) const { return subnetwork.get_final_data_gradient(); }
+
        const tensor& get_parameter_gradient(
        ) const { return details[0].get_parameter_gradient(); }