tf.raw_ops.MultiDeviceIteratorGetNextFromShard
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 .
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-01-23 UTC.
[null,null,["Last updated 2024-01-23 UTC."],[],[]]