Use the CPU softmax. Theano will move it to the GPU and cudnn softmax automatically if Theano can find cudnn:
http://deeplearning.net/software/theano/library/tensor/nnet/nnet.html?highlight=softmax#theano.tensor.nnet.nnet.softmax Fred On Mon, Aug 22, 2016 at 3:40 PM, Shir Gur <[email protected]> wrote: > Hi, > > > Softmax dim is usually 2. > > How should i implement my softmax with GpuDnnSoftmax? > > > thanks > > -- > > --- > You received this message because you are subscribed to the Google Groups > "theano-users" group. > To unsubscribe from this group and stop receiving emails from it, send an > email to [email protected]. > For more options, visit https://groups.google.com/d/optout. > -- --- You received this message because you are subscribed to the Google Groups "theano-users" group. To unsubscribe from this group and stop receiving emails from it, send an email to [email protected]. For more options, visit https://groups.google.com/d/optout.
