Esta página foi traduzida pela API Cloud Translation.
Switch to English

tf.RegisterGradient

TensorFlow 1 versão Ver fonte no GitHub

Um decorador para registrar a função de gradiente para um tipo op.

Este decorador é usado apenas quando se define um novo tipo op. Para um op com m entradas e n saídas, a função de inclinação é uma função que recebe o original Operation e n Tensor objetos (representando os gradientes com respeito a cada saída do op), e retorna m Tensor objetos (representando os gradientes parciais com respeitar para cada entrada do op).

Por exemplo, partindo do princípio de que as operações de tipo "Sub" ter duas entradas de x e y , e retornar uma saída única x - y , a função seguinte gradiente seriam registadas:

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

O argumento decorador op_type é o tipo de cadeia de uma operação. Isto corresponde ao OpDef.name campo para o proto que define a operação.

op_type O tipo de cadeia de uma operação. Isto corresponde ao OpDef.name campo para o proto que define a operação.

TypeError Se op_type não é string.

Métodos

__call__

Ver fonte

Registos a função f em função de gradiente para op_type .