- Notifications
You must be signed in to change notification settings - Fork1
A high level frontend Python machine learning framework/library for Tensor Array.
License
Tensor-Array/Tensor-Array-Python
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
This machine learning library usingTensor-Array library
This project is still in alpha version, we are trying to make this look like the main framework but it is easier to code.
Before install this library please installNVIDIA CUDA toolkit first.
It can not work withoutNVIDIA CUDA toolkit.
If you did not installPython then installPython 3:
apt-get updateapt-get install python3
After that go to command and install:
pip install TensorArray
Testing with theTensor object.
TheTensor class is a storage that store value and calculate the tensor.
TheTensor.calc_grad() method can do automatic differentiation.
TheTensor.get_grad() method can get the gradient after callTensor.calc_grad().
importtensor_array.coreastaimportnumpyasnpdeftest_add():example_tensor_array=ta.Tensor(np.array([ [1,2,3,4], [5,6,7,8], [9,10,11,12], [13,14,15,16] ],dtype=np.int32))example_tensor_array_scalar=ta.Tensor(100)example_tensor_sum=example_tensor_array+example_tensor_array_scalarprint(example_tensor_sum)example_tensor_sum.calc_grad()print(example_tensor_array.get_grad())print(example_tensor_array_scalar.get_grad())test_add()
About
A high level frontend Python machine learning framework/library for Tensor Array.
Topics
Resources
License
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Uh oh!
There was an error while loading.Please reload this page.