diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-10 23:29:49 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-10 23:30:06 -0700 |
commit | 7ed277ffd15aa0cf9eb8b65677e63ed54be17bb8 (patch) | |
tree | 34948eec1058ff622087b02e910de24947f70361 /tensorflow/contrib/layers | |
parent | 34ef46ca948440fa034c7b29cf1a516750eb02d3 (diff) | |
parent | d118516dd6c5b9fd2f0bfa2b870e7cfb5063e7dc (diff) |
Merge pull request #22031 from rogerxcn:master
PiperOrigin-RevId: 212399243
Diffstat (limited to 'tensorflow/contrib/layers')
-rw-r--r-- | tensorflow/contrib/layers/python/layers/layers.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/layers/python/layers/layers.py b/tensorflow/contrib/layers/python/layers/layers.py index 04668f112d..a82d4c1951 100644 --- a/tensorflow/contrib/layers/python/layers/layers.py +++ b/tensorflow/contrib/layers/python/layers/layers.py @@ -3109,7 +3109,7 @@ def maxout(inputs, num_units, axis=-1, scope=None): inputs: Tensor input num_units: Specifies how many features will remain after maxout in the `axis` dimension (usually channel). - This must be multiple of number of `axis`. + This must be a factor of number of features. axis: The dimension where max pooling will be performed. Default is the last dimension. scope: Optional scope for variable_scope. @@ -3128,7 +3128,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 |