org.tensorflow.op.data

Class MultiDeviceIteratorFromStringHandle.Options

java.lang.Object

org.tensorflow.op.data.MultiDeviceIteratorFromStringHandle.Options

Enclosing class:

MultiDeviceIteratorFromStringHandle

public static class MultiDeviceIteratorFromStringHandle.Options
extends Object

Optional attributes for MultiDeviceIteratorFromStringHandle

Method Summary

Modifier and Type	Method and Description
MultiDeviceIteratorFromStringHandle.Options	outputShapes(List<org.tensorflow.ndarray.Shape> outputShapes) Sets the outputShapes option.
MultiDeviceIteratorFromStringHandle.Options	outputShapes(org.tensorflow.ndarray.Shape... outputShapes) Sets the outputShapes option.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

outputShapes

public MultiDeviceIteratorFromStringHandle.Options outputShapes(List<org.tensorflow.ndarray.Shape> outputShapes)

Sets the outputShapes option.

Parameters:

outputShapes - The list of shapes being produced.

Returns:

this Options instance.

outputShapes

public MultiDeviceIteratorFromStringHandle.Options outputShapes(org.tensorflow.ndarray.Shape... outputShapes)

Sets the outputShapes option.

Parameters:

outputShapes - The list of shapes being produced.

Returns:

this Options instance.