1 #pragma once 2 3 #include <torch/csrc/Export.h> 4 #include <cstddef> 5 6 namespace torch { 7 namespace data { 8 namespace samplers { 9 /// A base class for custom index types. 10 struct TORCH_API CustomBatchRequest { 11 CustomBatchRequest() = default; 12 CustomBatchRequest(const CustomBatchRequest&) = default; 13 CustomBatchRequest(CustomBatchRequest&&) noexcept = default; 14 virtual ~CustomBatchRequest() = default; 15 16 /// The number of elements accessed by this index. 17 virtual size_t size() const = 0; 18 }; 19 } // namespace samplers 20 } // namespace data 21 } // namespace torch 22