cwenzi/neuroflow-cpp

1#include <iostream>2#include "../include/neuroflow/model.hpp"3 4using namespace neuroflow;5 6int main() {7    std::cout << "Simple model 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;  // 不使用 MLA18    19    std::cout << "Creating model..." << std::endl;20    NeuroFlowModel model(cfg);21    22    std::cout << "Creating input tensor..." << std::endl;23    Tensor input({1, cfg.input_dim});24    float* data = input.as_fp32();25    for (size_t i = 0; i < input.numel(); ++i) {26        data[i] = 0.1f * i;27    }28    29    std::cout << "Running forward..." << std::endl;30    auto output = model.forward(input);31    32    std::cout << "Output shape: [" << output.output.shape_[0] << ", " << output.output.shape_[1] << "]" << std::endl;33    std::cout << "Success!" << std::endl;34    35    return 0;36}37