TF定制优化函数

• 参考 https://github.com/tianheyu927/PCGrad/blob/master/PCGrad_tf.py
• 定制优化函数，考虑多任务学习 https://arxiv.org/pdf/2001.06782.pdf