提交 8b6144e7 authored 作者: Dustin Webb's avatar Dustin Webb

Added clarification comment.

上级 a684643a
...@@ -258,6 +258,8 @@ def test_softmax(): ...@@ -258,6 +258,8 @@ def test_softmax():
z = T.nnet.softmax z = T.nnet.softmax
f, f_gpu = _test_softmax(x, x, z, z, type(z), cuda.nnet.GpuSoftmax, cmp, -2) f, f_gpu = _test_softmax(x, x, z, z, type(z), cuda.nnet.GpuSoftmax, cmp, -2)
# cuDNN cannot handle these test cases but the Theano softmax can so we
# test them only for the Theano softmax.
cmp(2 << 15, 5, f, f_gpu) cmp(2 << 15, 5, f, f_gpu)
cmp(0, 10, f, f_gpu) cmp(0, 10, f, f_gpu)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论