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