Spec-Zone.ru › TensorFlow 2.4

tf.debugging.assert_shapes

Просмотреть исходный код на GitHub

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

tf.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: ...

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

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

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

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

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

Аргументы
shapes словарь с элементами (Tensor к форме) или список кортежей (Tensor, форма). Форма должна быть итерируемой.
data Тензоры для вывода, если условие ложно. По умолчанию отображаются сообщение об ошибке и первые несколько элементов нарушающего тензора.
summarize Количество выводимых элементов тензора.
message Строка, которая добавляется к стандартному сообщению.
name Имя данного оператора (необязательно). По умолчанию "assert_shapes".
Возбуждает
ValueError Если статические проверки покажут, что какое-либо ограничение формы нарушено.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/debugging/assert_shapes

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API