提交 893f9332 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

PEP8

上级 815b3049
...@@ -147,14 +147,17 @@ def verify_grad_sparse(op, pt, structured=False, *args, **kwargs): ...@@ -147,14 +147,17 @@ def verify_grad_sparse(op, pt, structured=False, *args, **kwargs):
Converts sparse variables back and forth. Converts sparse variables back and forth.
""" """
conv_none = lambda x: x conv_none = lambda x: x
def conv_csr(ind, indptr, shp): def conv_csr(ind, indptr, shp):
def f(spdata): def f(spdata):
return CSR(spdata, ind, indptr, shp) return CSR(spdata, ind, indptr, shp)
return f return f
def conv_csc(ind, indptr, shp): def conv_csc(ind, indptr, shp):
def f(spdata): def f(spdata):
return CSC(spdata, ind, indptr, shp) return CSC(spdata, ind, indptr, shp)
return f return f
iconv = [] iconv = []
dpt = [] dpt = []
...@@ -189,10 +192,12 @@ def verify_grad_sparse(op, pt, structured=False, *args, **kwargs): ...@@ -189,10 +192,12 @@ def verify_grad_sparse(op, pt, structured=False, *args, **kwargs):
oconv = DenseFromSparse(structured=structured) oconv = DenseFromSparse(structured=structured)
else: else:
oconv = conv_none oconv = conv_none
def conv_op(*inputs): def conv_op(*inputs):
ipt = [conv(i) for i, conv in zip(inputs, iconv)] ipt = [conv(i) for i, conv in zip(inputs, iconv)]
out = op(*ipt) out = op(*ipt)
return oconv(out) return oconv(out)
return utt.verify_grad(conv_op, dpt, *args, **kwargs) return utt.verify_grad(conv_op, dpt, *args, **kwargs)
verify_grad_sparse.E_grad = utt.verify_grad.E_grad verify_grad_sparse.E_grad = utt.verify_grad.E_grad
...@@ -746,7 +751,7 @@ class DenseFromSparse(gof.op.Op): ...@@ -746,7 +751,7 @@ class DenseFromSparse(gof.op.Op):
(self.sparse_grad == other.sparse_grad) (self.sparse_grad == other.sparse_grad)
def __hash__(self): def __hash__(self):
return hash(type(self))^hash(self.sparse_grad) return hash(type(self)) ^ hash(self.sparse_grad)
def __str__(self): def __str__(self):
return "%s{structured_grad=%s}" % ( return "%s{structured_grad=%s}" % (
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论