tensor_slices_dataset: Creates a dataset whose elements are slices of the given tensors.
Description
Creates a dataset whose elements are slices of the given tensors.
Usage
tensor_slices_dataset(tensors)
Arguments
tensors
A nested structure of tensors, each having the same size in
the 0th dimension.