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