提交 0a7fa210 authored 作者: affanv14's avatar affanv14

small fix for conv3d

上级 ec38f077
...@@ -2015,7 +2015,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer): ...@@ -2015,7 +2015,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer):
node.op.subsample, node.op.subsample,
node.op.filter_dilation) node.op.filter_dilation)
result[kshape] = theano.tensor.as_tensor_variable(node.op.kshp[-2:]) result[kshape] = theano.tensor.as_tensor_variable(node.op.kshp[2:])
for(var, shape) in zip((img, top), (node.op.imshp, tshp)): for(var, shape) in zip((img, top), (node.op.imshp, tshp)):
result[var] = theano.shared(np.random.random(shape).astype(var.dtype), result[var] = theano.shared(np.random.random(shape).astype(var.dtype),
...@@ -2032,7 +2032,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer): ...@@ -2032,7 +2032,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer):
node.op.subsample, node.op.subsample,
node.op.filter_dilation) node.op.filter_dilation)
result[ishape] = theano.tensor.as_tensor_variable(node.op.imshp[-2:]) result[ishape] = theano.tensor.as_tensor_variable(node.op.imshp[2:])
for(var, shape) in zip((kern, top), (node.op.kshp, tshp)): for(var, shape) in zip((kern, top), (node.op.kshp, tshp)):
result[var] = theano.shared(np.random.random(shape).astype(var.dtype), result[var] = theano.shared(np.random.random(shape).astype(var.dtype),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论