tf.nest.assert_same_structure

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

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

Обратите внимание, что кортежи с именами и полями с одинаковыми именами всегда считаются имеющими одинаковую поверхностную структуру (даже с check_types=True). Например, этот код выведет True:

def nt(a, b):
  return collections.namedtuple('foo', 'a b')(a, b)
print(assert_same_structure(nt(0, 1), nt(2, 3)))