rbm.hpp

#ifndef __RBM__HPP__
#define __RBM__HPP__

#include<cuv/basics/tensor.hpp>

namespace cuv{
namespace libs{
namespace rbm{

        template <class __value_type, class __memory_space_type, class __memory_layout_type>
        void set_binary_sequence(tensor<__value_type,__memory_space_type,__memory_layout_type>& m, const int& start);

        template <class __value_type, class __memory_space_type, class __memory_layout_type>
        void sigm_temperature(tensor<__value_type,__memory_space_type,__memory_layout_type>& m, const tensor<__value_type,__memory_space_type>& temp);

        template <class __value_type, class __memory_space_type, class __memory_layout_type>
        void set_local_connectivity_in_dense_matrix(tensor<__value_type,__memory_space_type,__memory_layout_type>& m, int patchsize, int vx, int vy, int hx, int hy, int maxdist_from_main_dia=1E6, bool round=false);



        template <class __value_type, class __memory_space_type, class __memory_layout_type>
        void copy_at_rowidx(tensor<__value_type,__memory_space_type,__memory_layout_type>& dst, const tensor<__value_type,__memory_space_type,__memory_layout_type>&  src, const tensor<typename tensor<__value_type,__memory_space_type,__memory_layout_type>::size_type,__memory_space_type,__memory_layout_type>& rowidx, const unsigned int offset);

        template <class __value_type, class __memory_space_type, class __memory_layout_type>
        void copy_redblack(tensor<__value_type,__memory_space_type,__memory_layout_type>& dst, const tensor<__value_type,__memory_space_type,__memory_layout_type>&  src, const unsigned int num_maps, const unsigned int color);

      template<class __value_type, class __memory_layout, class __memory_space_type>
              void bitflip(
              tensor<__value_type,__memory_layout,__memory_space_type> & matrix,
                              typename tensor<__value_type,__memory_layout,__memory_space_type>::size_type row);
} } }


#endif /* __RBM__HPP__ */