提交 33365fe6 authored 作者: affanv14's avatar affanv14

add docstring for GPUCorrMM

上级 0359b899
...@@ -487,6 +487,9 @@ class BaseGpuCorrMM(CGpuKernelBase): ...@@ -487,6 +487,9 @@ class BaseGpuCorrMM(CGpuKernelBase):
Perform subsampling of the output (default: (1, 1)). Perform subsampling of the output (default: (1, 1)).
filter_dilation filter_dilation
Perform subsampling of the input, also known as dilation (default: (1, 1)). Perform subsampling of the input, also known as dilation (default: (1, 1)).
num_groups :
Divides the image, kernel and output tensors into num_groups
separate groups. Each which carry out convolutions separately (default : 1).
""" """
check_broadcast = False check_broadcast = False
__props__ = ('border_mode', 'subsample', 'filter_dilation', 'num_groups') __props__ = ('border_mode', 'subsample', 'filter_dilation', 'num_groups')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论