tensor_dataset function

Dataset wrapping tensors.

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

tensor_dataset(...)

Arguments

• ...: tensors that have the same size of the first dimension.