tf.compat.v2.test.compute_gradient

Вычисляет теоретический и числовой якобиан f.

tf.compat.v2.test.compute_gradient(
    f, x, delta=0.001
)

При y = f(x), вычисляет теоретический и числовой якобиан dy/dx.

Аргументы
`f`	функция.
`x`	список аргументов для функции
`delta`	(необязательно) возмущение, используемое для вычисления числового якобиана.

Возвращает
Пара списков, где первый - список 2-мерных массивов NumPy, представляющих теоретические якобианы для каждого аргумента, а второй список - числовые. Каждый 2-мерный массив имеет "x_size" строк и "y_size" столбцов, где "x_size" - количество элементов в соответствующем аргументе, а "y_size" - количество элементов в f(x).

Возвращает

Пара списков, где первый - список 2-мерных массивов NumPy, представляющих теоретические якобианы для каждого аргумента, а второй список - числовые. Каждый 2-мерный массив имеет "x_size" строк и "y_size" столбцов, где "x_size" - количество элементов в соответствующем аргументе, а "y_size" - количество элементов в f(x).

Исключения
`ValueError`	Если результат пустой, но градиент не равен нулю.
`ValueError`	Если x не является списком, а имеет другой тип.

Пример:

@tf.function
def test_func(x):
  return x*x

theoretical, numerical = tf.test.compute_gradient(test_func, [1.0])
theoretical, numerical
# ((array([[2.]], dtype=float32),), (array([[2.000004]], dtype=float32),))

© 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/compat/v2/test/compute_gradient