tiny_dnn 1.0.0
A header only, dependency-free deep learning framework in C++11
Loading...
Searching...
No Matches
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
 
Uhost_at (const size_t d0, const size_t d1, const size_t d2, const size_t d3)
 
U host_at (const size_t d0, const size_t d1, const size_t d2, const size_t d3) const
 
const Uhost_ptr (const size_t d0, const size_t d1, const size_t d2, const size_t d3) const
 
Uhost_ptr (const size_t d0, const size_t d1, const size_t d2, const size_t d3)
 
const Uhost_data () const
 
Umutable_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: