Hi does the code in Daniel's answer run on GPU or CPU ?

import theano.tensor as tt

def threshold_activation(x):
    return tt.switch(tt.lt(x, 0), 0, 1)

I am looking for the same thing but I need to run every thing on GPU. 
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.

Reply via email to