未验证 提交 59e10922 编写于 作者: Q qingqing01 提交者: GitHub

Expose maxout Python API. (#11278)

* Expose maxout API.

* Fix code style.
上级 062d5a56
...@@ -73,6 +73,7 @@ __all__ = [ ...@@ -73,6 +73,7 @@ __all__ = [
'sum', 'sum',
'polygon_box_transform', 'polygon_box_transform',
'shape', 'shape',
'maxout',
] + __activations__ ] + __activations__
for _OP in set(__all__): for _OP in set(__all__):
......
...@@ -387,6 +387,14 @@ class TestBook(unittest.TestCase): ...@@ -387,6 +387,14 @@ class TestBook(unittest.TestCase):
self.assertIsNotNone(output) self.assertIsNotNone(output)
print(str(program)) print(str(program))
def test_maxout(self):
program = Program()
with program_guard(program):
data = layers.data(name='x', shape=[8, 6, 6], dtype="float32")
output = layers.maxout(x=data, groups=2)
self.assertIsNotNone(output)
print(str(program))
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册