cwenzi/neuroflow-cpp

1#include <iostream>2#include "../include/neuroflow/model.hpp"3 4using namespace neuroflow;5 6int main() {7    std::cout << "Simple forward test..." << std::endl;8    9    NeuroFlowModel::Config cfg;10    cfg.input_dim = 64;11    cfg.hidden_dim = 32;12    cfg.output_dim = 5;13    cfg.memory_slots = 8;14    cfg.memory_dim = 16;15    cfg.num_layers = 1;16    cfg.num_associations = 2;17    cfg.use_mla = false;18    19    NeuroFlowModel model(cfg);20    21    Tensor input({2, cfg.input_dim});22    for (size_t i = 0; i < input.numel(); ++i) input.as_fp32()[i] = 0.1f * i;23    24    std::cout << "Calling forward..." << std::endl;25    auto output = model.forward(input);26    27    std::cout << "Success!" << std::endl;28    return 0;29}30