tf.raw_ops.SymbolicGradient
Вычисляет функцию градиента для функции f с помощью обратного распространения.
tf.raw_ops.SymbolicGradient(
input, Tout, f, name=None
)
| Аргументы | |
|---|---|
input | Список объектов Tensor. список входных тензоров размером N + M; |
Tout | Список tf.DTypes, длина которого >= 1. список типов для входного списка. |
f | Функция, декорированная @Defun. Функция, для которой мы хотим вычислить градиент. Функция 'f' должна быть числовой функцией, которая принимает N входов и генерирует M выходов. Ее функция градиента 'g', которая вычисляется операцией SymbolicGradient, принимает N + M входов и генерирует N выходов. Например, если у нас есть (y1, y2, ..., y_M) = f(x1, x2, ..., x_N), то g - это (dL/dx1, dL/dx2, ..., dL/dx_N) = g(x1, x2, ..., x_N, dL/dy1, dL/dy2, ..., dL/dy_M), где L - скалярная функция от (x1, x2, ..., xN) (например, функция потерь). dL/dx_i - частная производная L по x_i. (Необходим специалист по математике, чтобы лучше сформулировать комментарий выше.) |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Список объектов Tensor типа Tout. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/SymbolicGradient