提交 15826072 authored 作者: Dustin Webb's avatar Dustin Webb

Changed variable from T.vector to T.fvector since the data will only be on a GPU.

上级 ff38034a
......@@ -385,7 +385,7 @@ class test_SoftMax(unittest.TestCase):
# Verify that the SoftmaxGrad -> GpuDnnSoftmaxGrad optimization is
# applied when cudnn is required
y = T.vector('y')
y = T.fvector('y')
f = theano.function(
[y],
T.grad(T.nnet.softmax(y).mean(), y),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论