tensor_slices_dataset function

Creates a dataset whose elements are slices of the given tensors.