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 в proto, определяющем операцию.

Методы

__call__

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

__call__(
    f: _T
) -> _T

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