提交 558854c4 authored 作者: David Warde-Farley's avatar David Warde-Farley

Python 2.4 compatibility fix.

上级 bdf1617d
...@@ -1450,8 +1450,11 @@ class StructuredDotGradCSR(gof.Op): ...@@ -1450,8 +1450,11 @@ class StructuredDotGradCSR(gof.Op):
} }
"""% dict(locals(), **sub) """% dict(locals(), **sub)
sdg_csr = StructuredDotGradCSR() sdg_csr = StructuredDotGradCSR()
class Dot(gof.op.Op): class Dot(gof.op.Op):
""" """
Operation for efficiently calculating the dot product when Operation for efficiently calculating the dot product when
...@@ -1505,12 +1508,17 @@ class Dot(gof.op.Op): ...@@ -1505,12 +1508,17 @@ class Dot(gof.op.Op):
def grad(self, (x, y), (gz,)): def grad(self, (x, y), (gz,)):
assert _is_sparse_variable(x) or _is_sparse_variable(y) assert _is_sparse_variable(x) or _is_sparse_variable(y)
rval = []
rval = [
tensor.dot(gz, y.T) if _is_dense_variable(y) else dot(gz, y.T), if _is_dense_variable(y):
tensor.dot(x.T, gz) if _is_dense_variable(x) else dot(x.T, gz) rval.append(tensor.dot(gz, y.T))
] else:
rval.append(dot(gz, y.T))
if _is_dense_variable(x):
rval.append(tensor.dot(x.T, gz))
else:
rval.append(dot(x.T, gz))
return rval return rval
_dot = Dot() _dot = Dot()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论