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

update conv2d doc about GPU.

上级 ebcc6a41
......@@ -61,15 +61,19 @@ def conv2d(input, filters, image_shape=None, filter_shape=None,
:param subsample: factor by which to subsample the output.
Also called strides elsewhere.
:type image_shape: None, tuple/list of len 4 of int or Constant variable
:type image_shape: None, tuple/list of len 4 of int, None or
Constant variable
:param image_shape: The shape of the input parameter.
Optional, used for optimization like loop unrolling
You can put None for any element of the list
to tell that this element is not constant.
:type filter_shape: None, tuple/list of len 4 of int or Constant variable
Not used on the GPU.
:type filter_shape: None, tuple/list of len 4 of int, None or
Constant variable
:param filter_shape: Optional, used for optimization like loop unrolling
You can put None for any element of the list
to tell that this element is not constant.
Not used on the GPU.
:param kwargs: kwargs are passed onto ConvOp.
Can be used to set the following:
unroll_batch, unroll_kern, unroll_patch,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论