Add spatial transformer to cuDNN documentation

上级 8fc10168
......@@ -146,6 +146,8 @@ To get an error if Theano can not use cuDNN, use this Theano flag:
- Softmax:
- You can manually use the op :class:`GpuDnnSoftmax
<theano.gpuarray.dnn.GpuDnnSoftmax>` to use its extra feature.
- Spatial Transformer:
- :func:`theano.gpuarray.dnn.dnn_spatialtf`.
List of Implemented Operations
==============================
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论