tf.contrib.training.create_train_op

Создаёт Operation, который вычисляет градиенты и возвращает значение потерь.

tf.contrib.training.create_train_op(
    total_loss, optimizer, global_step=_USE_GLOBAL_STEP, update_ops=None,
    variables_to_train=None, transform_grads_fn=None, summarize_gradients=False,
    gate_gradients=tf_optimizer.Optimizer.GATE_OP, aggregation_method=None,
    colocate_gradients_with_ops=False, check_numerics=True
)

Аргументы
`total_loss`	A `Tensor` представляющий общую потерю.
`optimizer`	tf.Optimizer для вычисления градиентов.
`global_step`	Переменная глобального шага. Если оставлено `_USE_GLOBAL_STEP`, используется tf.contrib.framework.global_step().
`update_ops`	Необязательный список обновлений. Если `update_ops` равно `None`, операторы обновления будут взяты из коллекции `tf.GraphKeys.UPDATE_OPS`. Если `update_ops` не равно `None`, но не содержит все операции обновления из `tf.GraphKeys.UPDATE_OPS`, будет выведено сообщение об ошибке.
`variables_to_train`	Необязательный список переменных для обучения. Если None, по умолчанию будут использоваться все tf.compat.v1.trainable_variables().
`transform_grads_fn`	Функция, принимающая список пар «градиент-переменная» (кортежи), выполняющая необходимые обновления градиентов (например, обрезку градиентов или умножение), и возвращающая обновлённый список.
`summarize_gradients`	Добавлять ли сводки для каждого градиента.
`gate_gradients`	Способ управления вычислением градиентов. Смотрите tf.Optimizer.
`aggregation_method`	Метод объединения градиентов. Допустимые значения определены в классе `AggregationMethod`.
`colocate_gradients_with_ops`	Попытка размещения градиентов вместе с операциями, которые их сгенерировали.
`check_numerics`	Применять ли check_numerics.

Возвращаемое значение
A `Tensor`, который при вычислении вычисляет градиенты и возвращает значение общей потери.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/training/create_train_op