|
nnlib
GPU-accelerated, C/C++ neural network library.
|
Header file declaring the Tensor class to represent multidimensional arrays. More...
#include "allocation.h"#include "cache.h"#include "session.cuh"#include <cstdlib>#include <iostream>#include <memory>#include <utility>#include <vector>Go to the source code of this file.
Classes | |
| class | Tensor |
| Class to represent multidimensional arrays. More... | |
Typedefs | |
| typedef std::shared_ptr< Tensor > | sTensor |
Functions | |
| std::ostream & | operator<< (std::ostream &stream, const Tensor &tensor) |
| Enables the tensor to be printed using std::cout. More... | |
| sTensor | sum (const sTensor &a) |
| void | fill (float value, sTensor &tensor) |
| void | fill (const sTensor &value, sTensor &tensor) |
| sTensor | add (const sTensor &a, const sTensor &b) |
| sTensor | subtract (const sTensor &a, const sTensor &b) |
| sTensor | hadamard (const sTensor &a, const sTensor &b) |
| sTensor | divide (const sTensor &a, const sTensor &b) |
| sTensor | log (const sTensor &a) |
| sTensor | multiply (const sTensor &a, float constant) |
| sTensor | multiply (const sTensor &a, const sTensor &b) |
| sTensor | transpose (const sTensor &a) |
| sTensor | relu (const sTensor &a) |
| sTensor | sigmoid (const sTensor &a) |
| sTensor | no_grad::sum (const sTensor &a) |
| sTensor | no_grad::add (const sTensor &a, const sTensor &b) |
| sTensor | no_grad::subtract (const sTensor &a, const sTensor &b) |
| sTensor | no_grad::hadamard (const sTensor &a, const sTensor &b) |
| sTensor | no_grad::divide (const sTensor &a, const sTensor &b) |
| sTensor | no_grad::log (const sTensor &a) |
| sTensor | no_grad::multiply (const sTensor &a, float constant) |
| sTensor | no_grad::multiply (const sTensor &a, const sTensor &b) |
| sTensor | no_grad::transpose (const sTensor &a) |
| sTensor | no_grad::relu (const sTensor &a) |
| sTensor | no_grad::sigmoid (const sTensor &a) |
Header file declaring the Tensor class to represent multidimensional arrays.
| std::ostream& operator<< | ( | std::ostream & | stream, |
| const Tensor & | tensor | ||
| ) |
Enables the tensor to be printed using std::cout.
| stream | The stream to print the tensor to. |
| tensor | The tensor to print. |