tiny_dnn  1.0.0
A header only, dependency-free deep learning framework in C++11
Public Member Functions | List of all members
tiny_dnn::Tensor< U > Class Template Reference

Public Member Functions

 Tensor (const size_t d0, const size_t d1, const size_t d2, const size_t d3)
 
 Tensor (const std::array< size_t, 4 > &shape)
 
 Tensor (const std::vector< size_t > &shape)
 
 Tensor (const Tensor &other)
 
Tensoroperator= (const Tensor &other)
 
 Tensor (Tensor &&other)
 
Tensoroperator= (Tensor &&other)
 
const std::array< size_t, 4 > & shape () const
 
U & host_at (const size_t d0, const size_t d1, const size_t d2, const size_t d3)
 
host_at (const size_t d0, const size_t d1, const size_t d2, const size_t d3) const
 
const U * host_ptr (const size_t d0, const size_t d1, const size_t d2, const size_t d3) const
 
U * host_ptr (const size_t d0, const size_t d1, const size_t d2, const size_t d3)
 
const U * host_data () const
 
U * mutable_host_data ()
 
size_t size () const
 
void fill (U value)
 
void reshape (const size_t d0, const size_t d1, const size_t d2, const size_t d3)
 
void reshape (const std::array< size_t, 4 > &sz)
 

The documentation for this class was generated from the following file: