提交 a1301828 authored 作者: Jesse Livezey's avatar Jesse Livezey

fix flake8

上级 9c15d602
...@@ -591,7 +591,7 @@ class test_DnnSoftMax(test_nnet.test_SoftMax): ...@@ -591,7 +591,7 @@ class test_DnnSoftMax(test_nnet.test_SoftMax):
xp = x.reshape((5, 5)) xp = x.reshape((5, 5))
y = T.nnet.softmax(xp.flatten()).sum() y = T.nnet.softmax(xp.flatten()).sum()
g = T.grad(y, x) g = T.grad(y, x)
f = theano.function(inputs=[x], outputs=g, mode=self.mode) theano.function(inputs=[x], outputs=g, mode=self.mode)
def test_cudnn_softmax_grad_opt(self): def test_cudnn_softmax_grad_opt(self):
# Verify that the SoftmaxGrad -> GpuDnnSoftmaxGrad optimization is # Verify that the SoftmaxGrad -> GpuDnnSoftmaxGrad optimization is
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论