tf.raw_ops.WindowDataset
Объединяет (вложенные) элементы входных данных в набор (вложенных) окон.
tf.raw_ops.WindowDataset(
input_dataset,
size,
shift,
stride,
drop_remainder,
output_types,
output_shapes,
metadata='',
name=None
)
«Окно» — это конечный набор плоских элементов размером size (или, возможно, меньшего, если нет достаточно элементов входных данных для заполнения окна и drop_remainder имеет значение false).
Аргумент shift определяет количество элементов входных данных, на которое сдвигается окно на каждой итерации. Первый элемент в k-ом окне будет элементом
1 + (k-1) * shift
набора входных данных. В частности, первый элемент первого окна всегда будет первым элементом входного набора данных.
Если параметр stride больше 1, то каждое окно будет пропускать (stride - 1) элементов входных данных между каждым элементом, который появляется в окне. Выходные окна по-прежнему будут содержать size элементов независимо от значения stride.
Аргумент stride определяет шаг элементов входных данных, а аргумент shift определяет смещение окна.
Например, если {...} представляет набор данных:
-
tf.data.Dataset.range(7).window(2)создает{ {0, 1}, {2, 3}, {4, 5}, {6} } -
tf.data.Dataset.range(7).window(3, 2, 1, True)создает{ {0, 1, 2}, {2, 3, 4}, {4, 5, 6} } -
tf.data.Dataset.range(7).window(3, 1, 2, True)создает{ {0, 2, 4}, {1, 3, 5}, {2, 4, 6} }
Обратите внимание, что при применении преобразования window к набору данных вложенных элементов оно создает набор данных вложенных окон.
Например:
-
tf.data.Dataset.from_tensor_slices((range(4), range(4))).window(2)создает{({0, 1}, {0, 1}), ({2, 3}, {2, 3})} -
tf.data.Dataset.from_tensor_slices({"a": range(4)}).window(2)создает{ {"a": {0, 1} }, {"a": {2, 3} } }
| Аргументы | |
|---|---|
input_dataset | Tensor типа variant. |
size | Tensor типа int64. Целочисленная скалярная величина, представляющая количество элементов входного набора данных, объединяемых в окно. Должно быть положительным. |
shift | Tensor типа int64. Целочисленная скалярная величина, представляющая количество элементов входных данных, на которое сдвигается окно на каждой итерации. По умолчанию size. Должно быть положительным. |
stride | Tensor типа int64. Целочисленная скалярная величина, представляющая шаг элементов входных данных в скользящем окне. Должно быть положительным. Значение по умолчанию 1 означает «сохранить каждый элемент входных данных». |
drop_remainder | Tensor типа bool. Булево скалярное значение, определяющее, следует ли отбрасывать последнее окно, если его размер меньше window_size. |
output_types | Список tf.DTypes длиной >= 1. |
output_shapes | Список форм (каждая — tf.TensorShape или список ints) длиной >= 1. |
metadata | Необязательный string. По умолчанию "". |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Tensor типа variant. |
© 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/versions/r2.9/api_docs/python/tf/raw_ops/WindowDataset