tf.compat.v1.assert_equal

Утверждает, что условие x == y выполняется поэлементно.

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.debugging.assert_equal

tf.compat.v1.assert_equal(
    x, y, data=None, summarize=None, message=None, name=None
)

Миграция на TF2

tf.compat.v1.assert_equal совместим с выполнением eager и tf.function. При миграции на TF2 используйте tf.debugging.assert_equal вместо этого. Помимо data, все аргументы поддерживаются с тем же именем аргумента.

Если вы хотите убедиться, что утверждения assert выполняются до потенциально неверного вычисления, используйте tf.control_dependencies, так как автоматические зависимости управления tf.function недостаточно для утверждений assert.

Структурное отображение на родной TF2

До:

tf.compat.v1.assert_equal(
  x=x, y=y, data=data, summarize=summarize,
  message=message, name=name)

После:

tf.debugging.assert_equal(
  x=x, y=y, message=message,
  summarize=summarize, name=name)

Пример использования TF1 и TF2

TF1:

g = tf.Graph()
with g.as_default():
  a = tf.compat.v1.placeholder(tf.float32, [2])
  b = tf.compat.v1.placeholder(tf.float32, [2])
  result = tf.compat.v1.assert_equal(a, b,
    message='"a == b" does not hold for the given inputs')
  with tf.compat.v1.control_dependencies([result]):
    sum_node = a + b
sess = tf.compat.v1.Session(graph=g)
val = sess.run(sum_node, feed_dict={a: [1, 2], b:[1, 2]})

TF2:

a = tf.Variable([1, 2], dtype=tf.float32)
b = tf.Variable([1, 2], dtype=tf.float32)
assert_op = tf.debugging.assert_equal(a, b, message=
  '"a == b" does not hold for the given inputs')
# When working with tf.control_dependencies
with tf.control_dependencies([assert_op]):
  val = a + b

Описание

Это условие выполняется, если для каждой пары (возможно, транслируемых) элементов x[i], y[i], мы имеем x[i] == y[i]. Если и x, и y пусты, это выполняется тривиально.

При выполнении в режиме графа вам следует добавить зависимость от этой операции, чтобы убедиться, что она выполняется. Пример добавления зависимости к операции:

with tf.control_dependencies([tf.compat.v1.assert_equal(x, y)]):
  output = tf.reduce_sum(x)

Аргументы
`x`	Числовой `Tensor`.
`y`	Числовой `Tensor`, того же типа и транслируемого к `x`.
`data`	Тензоры для вывода, если условие ложно. По умолчанию сообщение об ошибке и первые несколько элементов `x`, `y`.
`summarize`	Количество элементов каждого тензора для вывода.
`message`	Строка для добавления к сообщению по умолчанию.
`name`	Имя этой операции (необязательно). По умолчанию "assert_equal".

Возвращаемое значение
Операция, которая вызывает `InvalidArgumentError`, если `x == y` ложно.

Исключения
`InvalidArgumentError`	если проверка может быть выполнена немедленно, и `x == y` ложно. Проверка может быть выполнена немедленно во время выполнения eager или если `x` и `y` статически известны.

© 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/compat/v1/assert_equal