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