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

Fixes from code review

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