提交 a514bca1 authored 作者: Iban Harlouchet's avatar Iban Harlouchet

Correction (axis) to theano/compile/ops.py

上级 c4c90d32
...@@ -608,7 +608,7 @@ class Rebroadcast(gof.Op): ...@@ -608,7 +608,7 @@ class Rebroadcast(gof.Op):
def __init__(self, *axis): def __init__(self, *axis):
# Sort them to make sure we merge all possible case. # Sort them to make sure we merge all possible case.
items = sorted(iteritems(self.axis)) items = sorted(iteritems(axis))
self.axis = OrderedDict(items) self.axis = OrderedDict(items)
for axis, broad in iteritems(self.axis): for axis, broad in iteritems(self.axis):
assert isinstance(axis, (numpy.integer, int)), ( assert isinstance(axis, (numpy.integer, int)), (
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论