提交 9c0f3ef2 authored 作者: affanv14's avatar affanv14

add broadcastable argument while creating shared variable

上级 a65b0a11
......@@ -1800,7 +1800,9 @@ class ConvMetaOptimizer(LocalMetaOptimizer):
img, kern = node.inputs
for(var, shape) in zip((img, kern), shapes):
result[var] = theano.shared(np.random.random(shape).astype(var.dtype),
var.name, borrow=True)
var.name,
broadcastable=var.broadcastable,
borrow=True)
if type(node.op) in [AbstractConv2d_gradWeights, AbstractConv3d_gradWeights]:
img, top, kshape = node.inputs
......@@ -1815,7 +1817,9 @@ class ConvMetaOptimizer(LocalMetaOptimizer):
for(var, shape) in zip((img, top), (node.op.imshp, tshp)):
result[var] = theano.shared(np.random.random(shape).astype(var.dtype),
var.name, borrow=True)
var.name,
broadcastable=var.broadcastable,
borrow=True)
if type(node.op) in [AbstractConv2d_gradInputs, AbstractConv3d_gradInputs]:
kern, top, ishape = node.inputs
......@@ -1830,7 +1834,9 @@ class ConvMetaOptimizer(LocalMetaOptimizer):
for(var, shape) in zip((kern, top), (node.op.kshp, tshp)):
result[var] = theano.shared(np.random.random(shape).astype(var.dtype),
var.name, borrow=True)
var.name,
broadcastable=var.broadcastable,
borrow=True)
return result
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论