提交 cb07003d authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Fixes from code review

上级 45a631ee
......@@ -535,7 +535,7 @@ class Rebroadcast(gof.Op):
broadcast_pattern = []
else:
broadcast_pattern = ['?' for i
in xrange(1 + max(self.axis))]
in xrange(1 + numpy.max(self.axis.keys()))]
for k, v in self.axis.iteritems():
broadcast_pattern[k] = str(int(v))
return '%s{%s}' % (self.__class__.__name__,
......
......@@ -1498,7 +1498,7 @@ class SpSum(gof.op.Op):
def sp_sum(x, axis=None, sparse_grad=False):
"""Calculate the sum of a sparse matrix along a specific
"""Calculate the sum of a sparse matrix along the specified
axis.
It operates a reduction along the specified axis. When
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论