提交 ced46b7d authored 作者: carriepl's avatar carriepl 提交者: Frederic

Make dnn conv in gpuarray work with AbstractConv

上级 aafc23aa
......@@ -501,6 +501,14 @@ class GpuDnnConv(DnnBase):
or scalar.
"""
# if ishape and/or kshape are not tuples or list, but rather symbolic
# vectors, turn them into lists of symbolic scalars.
if not isinstance(ishape, (list, tuple)):
ishape = [ishape[i] for i in range(len(subsample) + 2)]
if not isinstance(kshape, (list, tuple)):
kshape = [kshape[i] for i in range(len(subsample) + 2)]
return get_conv_output_shape(
ishape,
kshape,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论