R/tensors_dataset.R

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

Arguments Description
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()