提交 203ac939 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Define 'bc01'.

上级 a58b05b9
...@@ -382,7 +382,8 @@ def dnn_conv(img, kerns, border_mode='valid', subsample=(1, 1), ...@@ -382,7 +382,8 @@ def dnn_conv(img, kerns, border_mode='valid', subsample=(1, 1),
""" """
GPU convolution using cuDNN from NVIDIA. GPU convolution using cuDNN from NVIDIA.
The memory layout to use is 'bc01'. The memory layout to use is 'bc01', that is 'batch', 'channel',
'first dim', 'second dim' in that order.
:param img: images to do the convolution over :param img: images to do the convolution over
:param kerns: convolution filters :param kerns: convolution filters
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论