R/zip_datasets.R

In rstudio/tfdatasets: Interface to 'TensorFlow' Datasets

#' Creates a dataset by zipping together the given datasets.
#'
#' Merges datasets together into pairs or tuples that contain an element from
#' each dataset.
#'
#' @param ... Datasets to zip (or a single argument with a list or list of lists of datasets).
#'
#' @return A dataset
#'
#' @export
zip_datasets <- function(...) {
  validate_tf_version()
  as_tf_dataset(
    tf$data$Dataset$zip(tuple(...))
  )
}