提交 171c015c authored 作者: Yann N. Dauphin's avatar Yann N. Dauphin

remove duplicated sparse sum op

上级 eeb70d82
...@@ -345,28 +345,6 @@ class EliminateZeros(gof.op.Op): ...@@ -345,28 +345,6 @@ class EliminateZeros(gof.op.Op):
eliminate_zeros = EliminateZeros() eliminate_zeros = EliminateZeros()
class Sum(gof.op.Op):
def __eq__(self, other):
return (type(self) == type(other))
def __hash__(self):
return hash(type(self))
def make_node(self, x, a):
x = as_sparse_variable(x)
a = tensor.as_tensor_variable(a)
return gof.Apply(self, [x, a], [tensor.TensorType(dtype=x.type.dtype,
broadcastable=(False,)).make_variable()])
def perform(self, node, (x, a), (out, )):
assert _is_sparse(x)
out[0] = numpy.asarray(x.sum(a), dtype=x.dtype).flatten()
def grad(self, (x, a, ), (gz, )):
return sp_ones_like(x) * gz, None
sum = Sum()
class Binomial(gof.op.Op): class Binomial(gof.op.Op):
def __init__(self, format, dtype): def __init__(self, format, dtype):
self.format = format self.format = format
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论