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

tf.recompute_grad

TensorFlow 1 versão Ver fonte no GitHub

Uma versão ansioso compatível de recompute_grad.

Para F (* args, kwargs **), Isto suporta gradientes com respeito a args ou kwargs, mas kwargs estão actualmente apenas suportado no modo ansioso. Note-se que para a camada e modelo Keras objetos, isso é feito automaticamente.

f função f(*x) que devolve um Tensor ou sequência de Tensor saídas.

Uma função de g que envoltórios f , mas que recalcula f nas trás passar de uma chamada de gradiente.