Program Listing for File custom_batch_request.h¶
↰ Return to documentation for file (torch/csrc/api/include/torch/data/samplers/custom_batch_request.h
)
#pragma once
#include <torch/csrc/Export.h>
#include <cstddef>
namespace torch {
namespace data {
namespace samplers {
struct TORCH_API CustomBatchRequest {
CustomBatchRequest() = default;
CustomBatchRequest(const CustomBatchRequest&) = default;
CustomBatchRequest(CustomBatchRequest&&) noexcept = default;
virtual ~CustomBatchRequest() = default;
virtual size_t size() const = 0;
};
} // namespace samplers
} // namespace data
} // namespace torch