logo头像

Welcome!May these help you.

custom_gradient

TF2.0自定义梯度函数

问题描述 在深度神经网络的训练过程中,经常需要自定义函数或一些变换,如果tf.gradient无法对自定义的函数进行自动求导计算梯度,那就需要对自定义的函数补充梯度计算函数。这里,我们以多输出神经网络结合自定义损失函数的例子展示如何使...