tf.debugging.assert_shapes
Проверка форм тензоров и соотношений размеров размерностей между тензорами.
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 | Если статическая проверка определит нарушение какого-либо ограничения формы. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/debugging/assert_shapes