提交 6b7844f2 authored 作者: Frederic's avatar Frederic

Add a note about the implementation that will be used.

上级 9d804c53
...@@ -159,6 +159,11 @@ def conv2d(input, ...@@ -159,6 +159,11 @@ def conv2d(input,
:rtype: symbolic 4D tensor :rtype: symbolic 4D tensor
:return: set of feature maps generated by convolutional layer. Tensor is :return: set of feature maps generated by convolutional layer. Tensor is
of shape (batch size, output channels, output rows, output columns) of shape (batch size, output channels, output rows, output columns)
:note: If CuDNN is available, it will be used on the
GPU. Otherwise, it is the *CorrMM* convolution that will be used
"caffe style convolution".
""" """
conv_op = AbstractConv2d(imshp=input_shape, conv_op = AbstractConv2d(imshp=input_shape,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论