diff --git a/dlib/dnn/layers.h b/dlib/dnn/layers.h index 5ac5aec01..06fd23b85 100644 --- a/dlib/dnn/layers.h +++ b/dlib/dnn/layers.h @@ -53,6 +53,43 @@ namespace dlib template using con = add_layer; +// ---------------------------------------------------------------------------------------- + + class bn_ + { + public: + bn_() + {} + + template + void setup (const SUBNET& sub) + { + // TODO + } + + template + void forward(const SUBNET& sub, resizable_tensor& output) + { + // TODO + } + + template + void backward(const tensor& gradient_input, SUBNET& sub, tensor& params_grad) + { + // TODO + } + + const tensor& get_layer_params() const { return params; } + tensor& get_layer_params() { return params; } + + private: + + resizable_tensor params; + }; + + template + using bn = add_layer; + // ---------------------------------------------------------------------------------------- class fc_