[nsthorat]

You can do something simple like this: https://github.com/tensorflow/tensorflow/blob/master/tensorf...

And a simple implementation: https://github.com/tensorflow/tensorflow/blob/master/tensorf...