Program Listing for File serialize.h¶
↰ Return to documentation for file (torch/csrc/api/include/torch/data/samplers/serialize.h
)
#pragma once
#include <torch/data/samplers/base.h>
#include <torch/serialize/archive.h>
namespace torch {
namespace data {
namespace samplers {
template <typename BatchRequest>
serialize::OutputArchive& operator<<(
serialize::OutputArchive& archive,
const Sampler<BatchRequest>& sampler) {
sampler.save(archive);
return archive;
}
template <typename BatchRequest>
serialize::InputArchive& operator>>(
serialize::InputArchive& archive,
Sampler<BatchRequest>& sampler) {
sampler.load(archive);
return archive;
}
} // namespace samplers
} // namespace data
} // namespace torch