tfdatasets (version 2.2.0)

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.

Value

A dataset.

See Also

Other tensor datasets: sparse_tensor_slices_dataset(), tensors_dataset()