用tensorflow实现