Skip to content

Latest commit

 

History

History
23 lines (20 loc) · 752 Bytes

cplusplus-front-end.md

File metadata and controls

23 lines (20 loc) · 752 Bytes
title order snippet
C++ Front-End
7
```cpp #include <torch/torch.h> torch::nn::Linear model(num_features, 1); torch::optim::SGD optimizer(model->parameters()); auto data_loader = torch::data::data_loader(dataset); for (size_t epoch = 0; epoch < 10; ++epoch) { for (auto batch : data_loader) { auto prediction = model->forward(batch.data); auto loss = loss_function(prediction, batch.target); loss.backward(); optimizer.step(); } } ```

The C++ frontend is a pure C++ interface to PyTorch that follows the design and architecture of the established Python frontend. It is intended to enable research in high performance, low latency and bare metal C++ applications.