Skip to content

Latest commit

 

History

History
30 lines (25 loc) · 2.11 KB

README.md

File metadata and controls

30 lines (25 loc) · 2.11 KB

NeuralNetwork

c++/asm implementation of neural network.

The layers works on type Tensor which represents $n$-dimensional array and supports mathematical operations (addition, subtraction, dot product, tensor product, $\ldots$) and other not math realted (adding padding, shuffling, reshaping $\ldots$). Some of the operations are optimized using AVX128 instructions, which made them a lot faster.

Example uses of NeuralNetwork class can be found in:

At this moment there are implemented layers of type:

  • dense,
  • activation (sigmoid, ReLU, leakyReLU),
  • conv2D (not optimized yet),
  • pool2D (max or mean pooling),
  • reshape,
  • flatten,
  • dropout,
  • normal distribution layer (for VAE).

Build scripts:

Other scripts: