Class MapAndBatchDataset.Inputs

java.lang.Object
org.tensorflow.op.RawOpInputs<MapAndBatchDataset>
org.tensorflow.op.data.MapAndBatchDataset.Inputs
Enclosing class:
MapAndBatchDataset
public static class MapAndBatchDataset.Inputs extends RawOpInputs<MapAndBatchDataset>

  • Field Details

    • inputDataset

      public final Operand<? extends TType> inputDataset
      A variant tensor representing the input dataset.

    • otherArguments

      public final Iterable<Operand<?>> otherArguments
      A list of tensors, typically values that were captured when building a closure for f.

    • batchSize

      public final Operand<TInt64> batchSize
      A scalar representing the number of elements to accumulate in a batch. It determines the number of concurrent invocations of f that process elements from input_dataset in parallel.

    • numParallelCalls

      public final Operand<TInt64> numParallelCalls
      A scalar representing the maximum number of parallel invocations of the map_fn function. Applying the map_fn on consecutive input elements in parallel has the potential to improve input pipeline throughput.

    • dropRemainder

      public final Operand<TBool> dropRemainder
      A scalar representing whether the last batch should be dropped in case its size is smaller than desired.

    • Targuments

      public final DataType[] Targuments
      The Targuments attribute

    • outputTypes

      public final DataType[] outputTypes
      The outputTypes attribute

    • outputShapes

      public final Shape[] outputShapes
      The outputShapes attribute

    • preserveCardinality

      public final boolean preserveCardinality
      The preserveCardinality attribute

    • metadata

      public final String metadata
      The metadata attribute

  • Constructor Details