Rate this Page
★★★★★
Struct TensorDataset#
Defined inFile tensor.h
Inheritance Relationships#
Base Type#
publictorch::data::datasets::Dataset<TensorDataset,TensorExample>(Template Class Dataset)
Struct Documentation#
- structTensorDataset:publictorch::data::datasets::Dataset<TensorDataset,TensorExample>#
A dataset of tensors.
Stores a single tensor internally, which is then indexed inside
get().Public Functions
- inlineexplicitTensorDataset(conststd::vector<Tensor>&tensors)#
Creates a
TensorDatasetfrom a vector of tensors.
- inlineexplicitTensorDataset(torch::Tensortensor)#
- inlinevirtualTensorExampleget(size_tindex)override#
Returns a single
TensorExample.
- inlinevirtualstd::optional<size_t>size()constoverride#
Returns the number of tensors in the dataset.
Public Members
- Tensortensor#
- inlineexplicitTensorDataset(conststd::vector<Tensor>&tensors)#
On this page
