使用GPU Tensorflow代码实现