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

Make dnn conv in gpuarray work with AbstractConv

上级 aafc23aa
...@@ -501,6 +501,14 @@ class GpuDnnConv(DnnBase): ...@@ -501,6 +501,14 @@ class GpuDnnConv(DnnBase):
or scalar. 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( return get_conv_output_shape(
ishape, ishape,
kshape, kshape,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论