#include #include #include #include #include using namespace std; torch::Tensor Normalize(torch::Tensor x) { x = x.div(255); x = x.sub(0.5).div(0.5); return x; } int main() { auto trainset = torch::data::datasets::MNIST("PATH_TO_STORE_TRAINSET") .map(Normalize); auto train_loader = torch::data::make_data_loadertorch::data::samplers::SequentialSampler( std::move(trainset), /batch_size=/64); class..