Binary Activation Function with Pytorch

[ptrblck]

This example could probably do, what you need.