From 88d3a0609be831561214911481bc5d73b3ceca53 Mon Sep 17 00:00:00 2001 From: Alexandre Passos Date: Mon, 20 Aug 2018 10:51:46 -0700 Subject: initializer might be a tensor so do not try to convert it to a boolean PiperOrigin-RevId: 209446309 --- tensorflow/python/layers/base.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'tensorflow/python/layers') 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, -- cgit v1.2.3