Skip to content

Commit 25c197e

Browse files
author
Michael Case
authored
Merge pull request tensorflow#20455 from lygztq/r1.9
fix bug in maxout function
2 parents 45fc11c + eb6c0cb commit 25c197e

File tree

1 file changed

+1
-1
lines changed
  • tensorflow/contrib/layers/python/layers

1 file changed

+1
-1
lines changed

tensorflow/contrib/layers/python/layers/layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3117,7 +3117,7 @@ def maxout(inputs, num_units, axis=-1, scope=None):
31173117
raise ValueError('number of features({}) is not '
31183118
'a multiple of num_units({})'.format(
31193119
num_channels, num_units))
3120-
shape[axis] = -1
3120+
shape[axis] = num_units
31213121
shape += [num_channels // num_units]
31223122

31233123
# Dealing with batches with arbitrary sizes

0 commit comments

Comments
 (0)