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  

Public Methods

Output<Object>
asOutput()
Returns the symbolic handle of a tensor.
static ShardDataset
create(Scope scope, Operand<?> inputDataset, Operand<Long> numShards, Operand<Long> index, List<Class<?>> 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

Public Methods

public Output<Object> asOutput ()

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.

public static ShardDataset create (Scope scope, Operand<?> inputDataset, Operand<Long> numShards, Operand<Long> index, List<Class<?>> 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)