tf.contrib.data.assert_element_shape

Проверить форму этого Dataset.

tf.contrib.data.assert_element_shape(
    expected_shapes
)

shapes = [tf.TensorShape([16, 256]), tf.TensorShape([None, 2])]
result = dataset.apply(tf.data.experimental.assert_element_shape(shapes))
print(result.output_shapes)  # ==> "((16, 256), (<unknown>, 2))"

Если формы набора данных и expected_shape полностью определены, проверить их соответствие. В противном случае добавить оператор проверки, который проверит формы при вычислении тензоров, и соответственно установить формы на тензоры.

Обратите внимание, что неизвестные размеры в expected_shapes будут проигнорированы.