我想替換或修改tensorflow中圖的運算或部分漸變。如果可以在計算中使用現有的梯度,那將是理想的。在某些方面,這與操作相反tf.stop_gradient():我希望添加僅在計算梯度時使用的計算,而不是添加在計算梯度時會忽略的計算。一個簡單的例子就是簡單地通過將梯度乘以常數來縮放比例(但不將正向計算乘以常數)。另一個例子是將梯度裁剪到給定范圍的東西。
添加回答
舉報
0/150
提交
取消