tf.compat.v1.debugging.assert_shapes

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

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

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

Пример:

n = 10
q = 3
d = 7
x = tf.zeros([n,q])
y = tf.ones([n,d])
param = tf.Variable([1.0, 2.0, 3.0])
scalar = 1.0
tf.debugging.assert_shapes([
 (x, ('N', 'Q')),
 (y, ('N', 'D')),
 (param, ('Q',)),
 (scalar, ()),
])

tf.debugging.assert_shapes([
  (x, ('N', 'D')),
  (y, ('N', 'D'))
])
Traceback (most recent call last):

ValueError: ...

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

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) или '*', это означает переменное число внешних измерений неопределённого размера, то есть ограничение применяется только к самым внутренним измерениям.

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