tf.RegisterGradient

Декоратор для регистрации функции градиента для типа операции.

tf.RegisterGradient(
    op_type
)

Этот декоратор используется только при определении нового типа операции. Для операции с m входами и n выходами функция градиента — это функция, которая принимает исходные Operation и n Tensor объекты (представляющие градиенты относительно каждого выхода операции) и возвращает m Tensor объекты (представляющие частные градиенты относительно каждого входа операции).

Например, предположим, что операции типа "Sub" принимают два входа x и y, и возвращают один выход x - y, следующая функция градиента будет зарегистрирована:

@tf.RegisterGradient("Sub")
def _sub_grad(unused_op, grad):
  return grad, tf.negative(grad)

Аргумент декоратора op_type — это строковый тип операции. Это соответствует полю OpDef.name в прото, определяющем операцию.

Методы

__call__

Просмотреть исходный код

__call__(
    f
)

Регистрирует функцию f как функцию градиента для op_type.