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...