tf.data.experimental.get_structure
Возвращает тип подписи для элементов входного набора данных/итератора.
tf.data.experimental.get_structure(
dataset_or_iterator
)
Например, чтобы получить структуру tf.data.Dataset:
dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3]) tf.data.experimental.get_structure(dataset) TensorSpec(shape=(), dtype=tf.int32, name=None)
dataset = tf.data.experimental.from_list([(1, 'a'), (2, 'b'), (3, 'c')]) tf.data.experimental.get_structure(dataset) (TensorSpec(shape=(), dtype=tf.int32, name=None), TensorSpec(shape=(), dtype=tf.string, name=None))
Чтобы получить структуру tf.data.Iterator:
dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3]) tf.data.experimental.get_structure(iter(dataset)) TensorSpec(shape=(), dtype=tf.int32, name=None)
| Аргументы | |
|---|---|
dataset_or_iterator | tf.data.Dataset или tf.data.Iterator. |
| Возвращаемое значение | |
|---|---|
(Вложенная) структура объектов tf.TypeSpec, соответствующая структуре элемента dataset_or_iterator и определяющая тип отдельных компонентов. |
| Исключения | |
|---|---|
TypeError | Если вход не является объектом tf.data.Dataset или tf.data.Iterator. |
© 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/data/experimental/get_structure