File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -14,7 +14,9 @@ Ptr<_Layer> _Layer::create() { \
14
14
15
15
Ptr<ReLULayer> ReLULayer::create (double negativeSlope)
16
16
{
17
- return Ptr<ReLULayer>(new ElementWiseLayer<ReLUFunctor>(ReLUFunctor (negativeSlope)));
17
+ Ptr<ReLULayer> layer (new ElementWiseLayer<ReLUFunctor>(ReLUFunctor (negativeSlope)));
18
+ layer->negativeSlope = negativeSlope;
19
+ return layer;
18
20
}
19
21
20
22
Ptr<TanHLayer> TanHLayer::create ()
@@ -40,7 +42,11 @@ Ptr<BNLLLayer> BNLLLayer::create()
40
42
Ptr<PowerLayer> PowerLayer::create (double power /* = 1*/ , double scale /* = 1*/ , double shift /* = 0*/ )
41
43
{
42
44
const PowerFunctor f (power, scale, shift);
43
- return Ptr<PowerLayer>(new ElementWiseLayer<PowerFunctor>(f));
45
+ Ptr<PowerLayer> layer (new ElementWiseLayer<PowerFunctor>(f));
46
+ layer->power = power;
47
+ layer->scale = scale;
48
+ layer->shift = shift;
49
+ return layer;
44
50
}
45
51
46
52
// //////////////////////////////////////////////////////////////////////////
You can’t perform that action at this time.
0 commit comments