TensorFlow 模型建立与训练

可以参考一下 https://www.tensorflow.org/api_docs/python/tf/custom_gradient