提交 9ef76c3f authored 作者: affanv14's avatar affanv14

add docs for gpucorr3dmm

上级 071e7fc9
......@@ -1067,6 +1067,9 @@ class BaseGpuCorr3dMM(CGpuKernelBase):
Perform subsampling of the output (default: (1, 1, 1)).
filter_dilation
Perform subsampling of the input, also known as dilation (default: (1, 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
......@@ -1495,6 +1498,11 @@ class GpuCorr3dMM(BaseGpuCorr3dMM):
The filter dilation operation applied to each input image.
Should be a tuple with 3 elements.
Set to `(1, 1, 1)` to disable filter dilation.
num_groups
The number of distinct groups the image and kernel must be
divided into.
should be an int
set to 1 to disable grouped convolution
Notes
-----
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论