cwenzi/neuroflow-cpp

1#include <iostream>2#include "../include/neuroflow/tensor.hpp"3#include "../include/neuroflow/networks.hpp"4 5using namespace neuroflow;6 7int main() {8    std::cout << "Linear detail test..." << std::endl;9    10    Linear linear(64, 32);11    12    Tensor input({2, 64});13    for (size_t i = 0; i < input.numel(); ++i) input.as_fp32()[i] = 0.1f * i;14    15    std::cout << "input shape: [" << input.shape_[0] << ", " << input.shape_[1] << "]" << std::endl;16    std::cout << "input numel: " << input.numel() << std::endl;17    std::cout << "input data_size: " << input.data_size_ << std::endl;18    std::cout << "input owns_data: " << input.owns_data_ << std::endl;19    20    std::cout << "Linear weight shape: [" << linear.weight.shape_[0] << ", " << linear.weight.shape_[1] << "]" << std::endl;21    std::cout << "Linear weight numel: " << linear.weight.numel() << std::endl;22    std::cout << "Linear weight data_size: " << linear.weight.data_size_ << std::endl;23    24    std::cout << "Calling linear.forward..." << std::endl;25    Tensor output = linear.forward(input);26    27    std::cout << "output shape: [" << output.shape_[0] << ", " << output.shape_[1] << "]" << std::endl;28    std::cout << "output numel: " << output.numel() << std::endl;29    std::cout << "output data_size: " << output.data_size_ << std::endl;30    std::cout << "output owns_data: " << output.owns_data_ << std::endl;31    32    // Check data pointer33    float* od = output.as_fp32();34    std::cout << "output data ptr: " << (void*)od << std::endl;35    std::cout << "First 5 output values: ";36    for (size_t i = 0; i < 5; ++i) std::cout << od[i] << " ";37    std::cout << std::endl;38    39    // Now do something with output40    std::cout << "Copying output..." << std::endl;41    Tensor copied = output.clone();42    43    std::cout << "Success!" << std::endl;44    return 0;45}46