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: _T
) -> _T

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