提交 0f4e3a8d authored 作者: Matthew Willson's avatar Matthew Willson

Fix get_conv_output_shape for symbolic input, removing a call to list() on a symbolic shape vector

上级 a7ac77a5
...@@ -43,8 +43,8 @@ def get_conv_output_shape(image_shape, kernel_shape, ...@@ -43,8 +43,8 @@ def get_conv_output_shape(image_shape, kernel_shape,
output channels, height and width of the image. None where undefined. output channels, height and width of the image. None where undefined.
""" """
bsize, imshp = image_shape[0], list(image_shape[2:]) bsize, imshp = image_shape[0], image_shape[2:]
nkern, kshp = kernel_shape[0], list(kernel_shape[2:]) nkern, kshp = kernel_shape[0], kernel_shape[2:]
if isinstance(border_mode, tuple): if isinstance(border_mode, tuple):
out_shp = tuple(get_conv_shape_1axis( out_shp = tuple(get_conv_shape_1axis(
imshp[i], kshp[i], border_mode[i], subsample[i]) imshp[i], kshp[i], border_mode[i], subsample[i])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论