org.tensorflow.op.core

Class ShardDataset

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.ShardDataset

All Implemented Interfaces:

Op, Operand<Object>

public final class ShardDataset
extends PrimitiveOp
implements Operand<Object>

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

Nested Class Summary

Nested Classes

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

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 ShardDataset	create(Scope scope, Operand<?> inputDataset, Operand<Long> numShards, Operand<Long> index, List<Class<?>> outputTypes, List<Shape> outputShapes, ShardDataset.Options... options) Factory method to create a class wrapping a new ShardDataset operation.
Output<?>	handle()
static ShardDataset.Options	requireNonEmpty(Boolean requireNonEmpty)

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

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

Method Detail

create

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

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

Parameters:

scope - current scope

inputDataset -

numShards - An integer representing the number of shards operating in parallel.

index - An integer representing the current worker index.

outputTypes -

outputShapes -

options - carries optional attributes values

Returns:

a new instance of ShardDataset

requireNonEmpty

public static ShardDataset.Options requireNonEmpty(Boolean requireNonEmpty)

Parameters:

requireNonEmpty -

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)