custom_batch_request.h

#pragma once

#include <cstddef>
#include <torch/csrc/WindowsTorchApiMacro.h>

namespace torch {
namespace data {
namespace samplers {
struct TORCH_API CustomBatchRequest {
  virtual ~CustomBatchRequest() = default;

  virtual size_t size() const = 0;
};
} // namespace samplers
} // namespace data
} // namespace torch