使用tensorflow实现