diff options
author | 2018-07-09 13:38:25 -0700 | |
---|---|---|
committer | 2018-07-09 13:38:25 -0700 | |
commit | 25c197e02393bd44f50079945409009dd4d434f8 (patch) | |
tree | 8c921bf3b46d092d819b0d55ec8e49821d84ec3d | |
parent | 45fc11c76b1e5755086432ea8bda6ae3300e542a (diff) | |
parent | eb6c0cbe460e6deafbe4252abed7e57bf9c4d133 (diff) |
Merge pull request #20455 from lygztq/r1.9
fix bug in maxout function
-rw-r--r-- | tensorflow/contrib/layers/python/layers/layers.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/layers/python/layers/layers.py b/tensorflow/contrib/layers/python/layers/layers.py index b7194ae333..a55d42c151 100644 --- a/tensorflow/contrib/layers/python/layers/layers.py +++ b/tensorflow/contrib/layers/python/layers/layers.py @@ -3117,7 +3117,7 @@ def maxout(inputs, num_units, axis=-1, scope=None): raise ValueError('number of features({}) is not ' 'a multiple of num_units({})'.format( num_channels, num_units)) - shape[axis] = -1 + shape[axis] = num_units shape += [num_channels // num_units] # Dealing with batches with arbitrary sizes |