Skip to content

yfengsong/Arsenal

 
 

Repository files navigation

Arsenal

  • Code For Interest

A computational Tool for Machine Learning

  • Long Time To be Finished
  • TT function added!

What we have:

  • Matrix Multiplication with optimization on single core
  • Common matrix Generation Function and shape function
  • Forward implemetation of tensor train based fully connected layers

TODO:

  • supporting convolutional layer
  • add parallel(openmp)
  • adjust Mat if needed
  • Backpropogation for fully connected layers.

About

Computational Lib

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 98.5%
  • Makefile 1.2%
  • Shell 0.3%