tf.debugging.assert_shapes

Проверка форм тензоров и соотношений размеров измерений между тензорами.

tf.debugging.assert_shapes(
    shapes, data=None, summarize=None, message=None, name=None
)

Этот оператор проверяет, что набор соотношений форм тензоров удовлетворяет заданным ограничениям.

Пример:

tf.assert_shapes({
  (x, ('N', 'Q')),
  (y, ('N', 'D')),
  (param, ('Q',)),
  (scalar, ())
})

Пример добавления зависимости к операции:

with tf.control_dependencies([tf.assert_shapes(shapes)]):
  output = tf.matmul(x, y, transpose_a=True)

Если x, y, param или scalar не имеют формы, удовлетворяющей всем указанным ограничениям, message, а также первые summarize элементы первого найденного нарушающего тензора будут напечатаны, и будет возбуждено исключение InvalidArgumentError.

Элементы размера в указанных формах проверяются по отношению к другим элементам по их хешу, за исключением:

• элемент размера интерпретируется как явный размер, если его можно разобрать как целочисленную примитивную переменную.
• элемент размера интерпретируется как любой размер, если он равен None или '.'.

Если первый элемент формы — ... (тип Ellipsis) или «*», это указывает на переменное число внешних измерений неопределенного размера, т. е. ограничение применяется только к внутренним измерениям.

Скалярные тензоры и указанные формы длиной ноль (исключая префикс «внутренние») оба рассматриваются как имеющие одно измерение размера один.