tf.grad_pass_through

Создаёт операцию grad-pass-through с поведением вперёд, заданным в f.

tf.grad_pass_through(
    f
)

Используйте эту функцию для обертывания любого оператора, сохраняя его поведение при прямом проходе, но заменяя исходный оператор в обратном графе на тождество. Например:

x = tf.Variable(1.0, name="x")
z = tf.Variable(3.0, name="z")

with tf.GradientTape() as tape:
  # y will evaluate to 9.0
  y = tf.grad_pass_through(x.assign)(z**2)
# grads will evaluate to 6.0
grads = tape.gradient(y, z)

Другой пример — «дифференцируемое» приближение скользящей средней, где градиенты разрешено передавать в последнее значение, подаваемое на скользящую среднюю, но скользящая средняя всё ещё используется для прямого прохода:

x = ... # Some scalar value
# A moving average object, we don't need to know how this is implemented
moving_average = MovingAverage()
with backprop.GradientTape() as tape:
  # mavg_x will evaluate to the current running average value
  mavg_x = tf.grad_pass_through(moving_average)(x)
grads = tape.gradient(mavg_x, x) # grads will evaluate to 1.0