nnlib/main.cpp at main · rosxnb/nnlib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#include "nn.hpp"

int main()
{
    std::cout.setf(std::ios::fixed);

    Mat m(2, 2);
    Mat n(2, 2);

    m.fill(1.f);
    n.fill(1.f);
    std::cout << m << std::endl;
    std::cout << n;

    m += n;
    std::cout << "---------------------------------------------------------------------------\n";
    std::cout << m << std::endl;
    std::cout << "---------------------------------------------------------------------------\n";

    Mat a(4, 4, 1.f);
    std::cout << a << std::endl;

    Mat b(4, 2);
    b.randomise();
    std::cout << b;

    std::cout << "---------------------------------------------------------------------------\n";
    Mat res = a * b;
    std::cout << res;
    std::cout << "---------------------------------------------------------------------------\n";

    return 0;
}