tensor_dataset: Dataset wrapping tensors.

Description Usage Arguments

Description

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

Usage

1

Arguments

...

tensors that have the same size of the first dimension.


torch documentation built on Oct. 7, 2021, 9:22 a.m.