ShardDataset

public final class ShardDataset

Creates a `Dataset` that includes only 1/`num_shards` of this dataset.

Nested Classes

class ShardDataset.Options Optional attributes for ShardDataset  

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

Output<TType>
asOutput()
Returns the symbolic handle of the tensor.
static ShardDataset
create(Scope scope, Operand<?> inputDataset, Operand<TInt64> numShards, Operand<TInt64> index, List<Class<? extends TType>> outputTypes, List<Shape> outputShapes, Options... options)
Factory method to create a class wrapping a new ShardDataset operation.
Output<?>
handle()
static ShardDataset.Options
requireNonEmpty(Boolean requireNonEmpty)

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "ShardDataset"

Public Methods

public Output<TType> asOutput ()

Returns the symbolic handle of the 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.

public static ShardDataset create (Scope scope, Operand<?> inputDataset, Operand<TInt64> numShards, Operand<TInt64> index, List<Class<? extends TType>> outputTypes, List<Shape> outputShapes, Options... options)

Factory method to create a class wrapping a new ShardDataset operation.

Parameters
scope current scope
numShards An integer representing the number of shards operating in parallel.
index An integer representing the current worker index.
options carries optional attributes values
Returns
  • a new instance of ShardDataset

public Output<?> handle ()

public static ShardDataset.Options requireNonEmpty (Boolean requireNonEmpty)