Gets next element for the provided shard number.
tf.raw_ops.MultiDeviceIteratorGetNextFromShard(
multi_device_iterator,
shard_num,
incarnation_id,
output_types,
output_shapes,
name=None
)
Args |
multi_device_iterator
|
A Tensor of type resource .
A MultiDeviceIterator resource.
|
shard_num
|
A Tensor of type int32 .
Integer representing which shard to fetch data for.
|
incarnation_id
|
A Tensor of type int64 .
Which incarnation of the MultiDeviceIterator is running.
|
output_types
|
A list of tf.DTypes that has length >= 1 .
The type list for the return values.
|
output_shapes
|
A list of shapes (each a tf.TensorShape or list of ints ) that has length >= 1 .
The list of shapes being produced.
|
name
|
A name for the operation (optional).
|
Returns |
A list of Tensor objects of type output_types .
|