Creates a Dataset
that includes only 1/num_shards
of this dataset.
View aliases
Compat aliases for migration
See Migration guide for more details.
tf.raw_ops.ShardDataset(
input_dataset,
num_shards,
index,
output_types,
output_shapes,
require_non_empty=False,
metadata='',
name=None
)
Args | |
---|---|
input_
|
A Tensor of type variant .
|
num_
|
A Tensor of type int64 .
An integer representing the number of shards operating in parallel.
|
index
|
A Tensor of type int64 .
An integer representing the current worker index.
|
output_
|
A list of tf. that has length >= 1 .
|
output_
|
A list of shapes (each a tf.TensorShape or list of ints ) that has length >= 1 .
|
require_
|
An optional bool . Defaults to False .
|
metadata
|
An optional string . Defaults to "" .
|
name
|
A name for the operation (optional). |
Returns | |
---|---|
A Tensor of type variant .
|