tf.nest.assert_same_structure

Проверяет, что две структуры вложены одинаковым образом.

tf.nest.assert_same_structure(
    nest1, nest2, check_types=True, expand_composites=False
)

Обратитесь к tf.nest для определения структуры.

Обратите внимание, что метод не проверяет типы атомов внутри структур.

Примеры:

• Эти сравнения атом-атом пройдут:

tf.nest.assert_same_structure(1.5, tf.Variable(1, tf.uint32))
tf.nest.assert_same_structure("abc", np.array([1, 2]))
  

• Эти сравнения вложенных структур пройдут:

structure1 = (((1, 2), 3), 4, (5, 6))
structure2 = ((("foo1", "foo2"), "foo3"), "foo4", ("foo5", "foo6"))
structure3 = [(("a", "b"), "c"), "d", ["e", "f"]]
tf.nest.assert_same_structure(structure1, structure2)
tf.nest.assert_same_structure(structure1, structure3, check_types=False)
  

import collections
tf.nest.assert_same_structure(
    collections.namedtuple("bar", "a b")(1, 2),
    collections.namedtuple("foo", "a b")(2, 3),
    check_types=False)
  

tf.nest.assert_same_structure(
    collections.namedtuple("bar", "a b")(1, 2),
    { "a": 1, "b": 2 },
    check_types=False)
  

tf.nest.assert_same_structure(
    { "a": 1, "b": 2, "c": 3 },
    { "c": 6, "b": 5, "a": 4 })
  

ragged_tensor1 = tf.RaggedTensor.from_row_splits(
      values=[3, 1, 4, 1, 5, 9, 2, 6],
      row_splits=[0, 4, 4, 7, 8, 8])
ragged_tensor2 = tf.RaggedTensor.from_row_splits(
      values=[3, 1, 4],
      row_splits=[0, 3])
tf.nest.assert_same_structure(
      ragged_tensor1,
      ragged_tensor2,
      expand_composites=True)
  

• Эти примеры приведут к исключениям:

tf.nest.assert_same_structure([0, 1], np.array([0, 1]))
  Traceback (most recent call last):

  ValueError: The two structures don't have the same nested structure
  

tf.nest.assert_same_structure(
      collections.namedtuple('bar', 'a b')(1, 2),
      collections.namedtuple('foo', 'a b')(2, 3))
  Traceback (most recent call last):

  TypeError: The two structures don't have the same nested structure