org.tensorflow.op.core

Class ShuffleDataset

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.ShuffleDataset

All Implemented Interfaces:

Op, Operand<Object>

public final class ShuffleDataset
extends PrimitiveOp
implements Operand<Object>

Creates a dataset that shuffles elements from `input_dataset` pseudorandomly.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ShuffleDataset.Options Optional attributes for ShuffleDataset

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
Output<Object>	asOutput() Returns the symbolic handle of a tensor.
static ShuffleDataset	create(Scope scope, Operand<?> inputDataset, Operand<Long> bufferSize, Operand<Long> seed, Operand<Long> seed2, List<Class<?>> outputTypes, List<Shape> outputShapes, ShuffleDataset.Options... options) Factory method to create a class to wrap a new ShuffleDataset operation to the graph.
Output<?>	handle()
static ShuffleDataset.Options	reshuffleEachIteration(Boolean reshuffleEachIteration)

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static ShuffleDataset create(Scope scope,
                                    Operand<?> inputDataset,
                                    Operand<Long> bufferSize,
                                    Operand<Long> seed,
                                    Operand<Long> seed2,
                                    List<Class<?>> outputTypes,
                                    List<Shape> outputShapes,
                                    ShuffleDataset.Options... options)

Factory method to create a class to wrap a new ShuffleDataset operation to the graph.

Parameters:

scope - current graph scope

inputDataset -

bufferSize - The number of output elements to buffer in an iterator over this dataset. Compare with the `min_after_dequeue` attr when creating a `RandomShuffleQueue`.

seed - A scalar seed for the random number generator. If either `seed` or `seed2` is set to be non-zero, the random number generator is seeded by the given seed. Otherwise, a random seed is used.

seed2 - A second scalar seed to avoid seed collision.

outputTypes -

outputShapes -

options - carries optional attributes values

Returns:

a new instance of ShuffleDataset

reshuffleEachIteration

public static ShuffleDataset.Options reshuffleEachIteration(Boolean reshuffleEachIteration)

Parameters:

reshuffleEachIteration - If true, each iterator over this dataset will be given a different pseudorandomly generated seed, based on a sequence seeded by the `seed` and `seed2` inputs. If false, each iterator will be given the same seed, and repeated iteration over this dataset will yield the exact same sequence of results.

handle

public Output<?> handle()

asOutput

public Output<Object> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

Specified by:

asOutput in interface Operand<Object>

See Also:

OperationBuilder.addInput(Output)