提交 413531a2 authored 作者: Frederic's avatar Frederic

add docstring and swap parameter order to be the same as make_node

上级 38ebc17f
......@@ -418,7 +418,14 @@ def dnn_conv(img, kerns, border_mode='valid', subsample=(1, 1),
class GpuDnnPoolDesc(GpuOp):
__props__ = ('mode', 'ws', 'stride')
"""
The pooling descriptor.
:param ws: windows size
:param stride: (dx, dy)
:param mode: 'max' or 'average'
"""
__props__ = ('ws', 'stride', 'mode')
def c_headers(self):
return ['cudnn.h', 'cudnn_helper.h']
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论