未验证 提交 25c197e0 编写于 作者: M Michael Case 提交者: GitHub

Merge pull request #20455 from lygztq/r1.9

fix bug in maxout function
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册