Eample of using dataset api in tensorflow - View it on GitHub