Class HostDnTen

XM/include/Utils/memory.h:189–273

template <typename T>
class HostDnTen {
public:
    size_s* dimensions = nullptr; // Array to store the size of each dimension
    size_s num_dims = 0;    // Number of dimensions
    size_l total_size = 1;
    T* vals = nullptr;         // Pointer to the array of values
    T initial_value;
    bool is_initial_value = false;

    HostDnTen() {}
    
    HostDnTen(const size_s num_dims, const size_s* dimensions) : num_dims(num_dims){
        this->dimensions = (size_s*)malloc(sizeof(size_s) * num_dims);
        std::memcpy(this->dimensions, dimensions, sizeof(size_s) * num_dims); // Copy dimensions
        total_size = 1;
        for (size_s i = 0; i < num_dims; ++i) {
            total_size *= dimensions[i]; // Calculate total number of elements
        }
        this->allocate();
    }
    HostDnTen(const size_s num_dims, const size_s* dimensions, T initial_value) : num_dims(num_dims) ,initial_value(initial_value){
        this->dimensions = (size_s*)malloc(sizeof(size_s) * num_dims);
        std::memcpy(this->dimensions, dimensions, sizeof(size_s) * num_dims); // Copy dimensions
        is_initial_value = true;
        total_size = 1;
        for (size_s i = 0; i < num_dims; ++i) {
            total_size *= dimensions[i]; // Calculate total number of elements
        }
        this->allocate();
    }

    HostDnTen(std::initializer_list<size_s> dims) : num_dims(dims.size()){
        dimensions = new size_s[num_dims];
        std::copy(dims.begin(), dims.end(), dimensions);
        total_size = 1;
        for (size_s i = 0; i < num_dims; ++i) {
            total_size *= dimensions[i]; // Calculate total number of elements
        }
        this->allocate();
    }

    // Allocate memory for the values
    void allocate() {
        if (this->vals == nullptr) {
            this->vals = (T*)malloc(sizeof(T) * total_size);
            if (this->vals && is_initial_value) {
                std::fill(this->vals, this->vals + total_size, initial_value);
            }
        }else{
            std::cout << "Error: Vec already allocated!" << std::endl;
        }
    }

    // Print matrix, if matrix is larger the 3-dimension then donot print
    void print() {
        if(num_dims == 1){
            for (size_s j = 0; j < dimensions[0]; j++) {
                std::printf("%0.2f ", vals[j]);