diff options
Diffstat (limited to 'tensorflow/python/layers/base.py')
-rw-r--r-- | tensorflow/python/layers/base.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/layers/base.py b/tensorflow/python/layers/base.py index ab08865532..3ba880d7a1 100644 --- a/tensorflow/python/layers/base.py +++ b/tensorflow/python/layers/base.py @@ -262,11 +262,13 @@ class Layer(base_layer.Layer): use_resource = (use_resource or self._use_resource_variables or scope.use_resource) + if initializer is None: + initializer = scope.initializer variable = super(Layer, self).add_weight( name, shape, dtype=dtypes.as_dtype(dtype), - initializer=initializer or scope.initializer, + initializer=initializer, trainable=trainable, constraint=constraint, partitioner=partitioner, |