Dataset

Группирует элементы этого набора данных в пакеты.

Группирует элементы этого набора данных в пакеты. Включает последний пакет, даже если он содержит меньше элементов `batchSize`.

Создает в памяти набор данных, элементами которого являются фрагменты заданных тензоров. Каждый элемент этого набора данных будет List<Operand<?>> , представляющий фрагменты (например, пакеты) предоставленных тензоров.

Получите список фигур для каждого компонента этого набора данных.

Получите список типов вывода для каждого компонента этого набора данных.

Получите вариант тензора, представляющий этот набор данных.

Создает итератор, который быстро перебирает все пакеты этого набора данных. Каждый пакет представляет собой список компонентов, возвращаемых как объекты «Выход».

Этот метод позволяет выполнять каждую итерацию по пакетам при работе в активном режиме. Информацию о пакетной итерации в режиме графика см. в `makeOneShotIterator`.

Создает DatasetIterator, который можно использовать для перебора элементов этого набора данных.

Этот итератор должен быть инициализирован вызовом `iterator.makeInitializer(Dataset)`, прежде чем элементы можно будет получить в цикле.

Создает DatasetIterator, который можно использовать для перебора элементов этого набора данных. Использование makeOneShotIterator гарантирует, что итератор автоматически инициализируется в этом наборе данных. пропуски. В режиме графа операция инициализации будет добавлена ​​в список инициализаторов графа, который необходимо запустить через `tf.init()`:

Бывший:

     try (Session session = new Session(graph) {
         // Immediately run initializers
         session.run(tf.init());
     }
 

В режиме ожидания инициализатор будет запущен автоматически в результате этого вызова.

Возвращает новый набор данных, который отображает функцию на все элементы, возвращаемые этим набором данных.

Например, предположим, что каждый элемент представляет собой List<Operand<?>> с двумя компонентами: (функции, метки).

Вызов

dataset.map(components -> {
      Operand<?> features = components.get(0);
      Operand<?> labels   = components.get(1);

      return Arrays.asList(
        tf.math.mul(features, tf.constant(2)),
        tf.math.mul(labels, tf.constant(5))
      );
 );
 }

Возвращает новый набор данных, который сопоставляет функцию всем элементам этого набора данных, всем компонентам каждого элемента.

Например, предположим, что каждый элемент представляет собой List<Operand<?>> с двумя компонентами: (функции, метки).

Вызов dataset.mapAllComponents(component -> tf.math.mul(component, tf.constant(2))) сопоставит функцию с компонентами `features` и `labels` каждого элемента, умножив их все на 2.

Возвращает новый набор данных, который сопоставляет функцию всем элементам этого набора данных с одним компонентом каждого элемента.

Например, предположим, что каждый элемент представляет собой List<Operand<?>> с двумя компонентами: (функции, метки).

Вызов dataset.mapOneComponent(0, features -> tf.math.mul(features, tf.constant(2))) сопоставит функцию с компонентом `features` каждого элемента, умножив каждый на 2.

Возвращает новый набор данных, который пропускает начальные элементы count из этого набора данных.

Возвращает новый набор данных, содержащий только первые элементы count из этого набора данных.