stan_was_stupid.py

gistfile1.txt

import theano
import theano.tensor as T
import keras.backend as K
import numpy as np
x = T.fmatrix()
mask = T.ones_like(x[:, 0])
mask = mask.dimshuffle(0, "x")
mask = K.dropout(mask, 0.5)
fnc = theano.function([x], mask)
print fnc(np.array([[1,2,3.], [4,5,6], [7,8,9]]).astype("float32"))
print fnc(np.array([[1,2,3.], [4,5,6], [7,8,9]]).astype("float32"))
print fnc(np.array([[1,2,3.], [4,5,6], [7,8,9]]).astype("float32"))

"""
Outputs:

[[ 2.]
 [ 0.]
 [ 2.]]
[[ 2.]
 [ 0.]
 [ 2.]]
[[ 0.]
 [ 0.]
 [ 2.]]

"""