tiny_dnn
1.0.0
A header only, dependency-free deep learning framework in C++11
|
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) | |
Tensor & | operator= (const Tensor &other) |
Tensor (Tensor &&other) | |
Tensor & | operator= (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) |
U | 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) |