tensor_dataset: Dataset wrapping tensors.

tensor_datasetR Documentation

Dataset wrapping tensors.

Description

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.


torch documentation built on Aug. 21, 2025, 5:50 p.m.