Dataset wrapping tensors.

Each sample will be retrieved by indexing tensors along the first dimension.

Usage

tensor_dataset(...)

Arguments

...

tensors that have the same size of the first dimension.