Function tensorflow::ops::rebatch_dataset_v2
source · pub fn rebatch_dataset_v2<O0: Into<Output>, O1: Into<Output>, O2: Into<Output>>(
input_dataset: O0,
batch_sizes: O1,
drop_remainder: O2,
scope: &mut Scope
) -> Result<Operation>
Expand description
Shorthand for RebatchDatasetV2::new().build(input_dataset, batch_sizes, drop_remainder, scope)
.