A tensorflow implementation for a differentiable stochastic estimator of the information transmitted by a layer of binary neurons. Requires tensorflow >= 1.3 and all dependencies therein.
jaberkow/stochastic_I
A tensorflow implementation for a differentiable stochastic estimator of the information transmitted by a layer of binary neurons
Jupyter Notebook