tf.raw_ops.ParallelBatchDataset
tf.raw_ops.ParallelBatchDataset(
input_dataset,
batch_size,
num_parallel_calls,
drop_remainder,
output_types,
output_shapes,
parallel_copy=False,
deterministic='default',
metadata='',
name=None
)
Args |
input_dataset |
A Tensor of type variant . |
batch_size |
A Tensor of type int64 . |
num_parallel_calls |
A Tensor of type int64 . |
drop_remainder |
A Tensor of type bool . |
output_types |
A list of tf.DTypes that has length >= 1 . |
output_shapes |
A list of shapes (each a tf.TensorShape or list of ints ) that has length >= 1 . |
parallel_copy |
An optional bool . Defaults to False . |
deterministic |
An optional string . Defaults to "default" . |
metadata |
An optional string . Defaults to "" . |
name |
A name for the operation (optional). |
Returns |
A Tensor of type variant . |