|
|
using | BatchType = std::vector< int > |
| |
|
using | ChunkType = std::vector< int > |
| |
|
|
BatchType | read_chunk (size_t chunk_index) override |
| | Read an entire chunk.
|
| |
|
size_t | chunk_count () override |
| | Returns the number of chunks available in this reader.
|
| |
|
void | reset () override |
| | This will clear any internal state associate with this reader.
|
| |
|
|
size_t | chunk_sizes [chunk_count_] = {10, 5, 20} |
| |
|
|
static const size_t | chunk_count_ = 3 |
| |
Definition at line 65 of file dataloader.cpp.
The documentation for this struct was generated from the following file:
Public Types inherited from torch::data::datasets::ChunkDataReader< std::vector< int > >
1.8.11
