/TensorFly

A project for SJTU PPCA 2019 ML system.

Primary LanguagePython

TensorFly

TensorFly is an open source project (for SJTU PPCA 2019 ML-system). Its API is almost the same as TensorFlow (r1.14 stable), but many methods are not implemented.

How to run test cases

Testcases are here.

  1. Download numpy.

  2. Download Intel Math Kernel Library. You can follow the instructions here.

  3. To compile core.c:

    gcc -o tensorfly/core.so -shared -fPIC -fopenmp tensorfly/core.c -lmkl_rt -O4 -funroll-loops

  4. Run test:

    python3 run_test.py tensorfly