unaryfunction.h

Go to the documentation of this file.

#ifndef BLACKCAT_NEURALNETWORK_UNARYFUNCTION_H_
#define BLACKCAT_NEURALNETWORK_UNARYFUNCTION_H_

#include "layer_base.h"
#include <string>

namespace bc {
namespace nn {

template<
    class SystemTag,
    class ValueType,
    class Functor,
    class InputDimension = bc::traits::Integer<1>>
struct Function:
    Layer_Base<
        Function<SystemTag, ValueType, Functor, InputDimension>,
        Tensor_Descriptor<ValueType, SystemTag, InputDimension>>
{
    using system_tag = SystemTag;
    using value_type = ValueType;
    using allocator_type = nn_default_allocator_type<ValueType, SystemTag>;
    using self_type = Function<SystemTag, ValueType, Functor,InputDimension>;

    using input_descriptor_t = Tensor_Descriptor<ValueType, SystemTag, InputDimension>;
    using parent_type = Layer_Base<self_type, input_descriptor_t>;
    using input_tensor_dim = InputDimension;
    using output_tensor_dim = InputDimension;

    using typename parent_type::shape_type;

    Functor function;

    Function(shape_type inputs, Functor function=Functor()):
        parent_type(bc_get_classname_of(function), inputs, inputs),
        function(function) {}

    template<class Matrix>
    auto forward_propagation(const Matrix& x) {
        return function(x);
    }

    template<class X, class Delta>
    auto back_propagation(const X& x, const Delta& dy) {
        return function.dx(x) % dy;
    }
};


template<class ValueType, class SystemTag, class Functor>
Function<SystemTag, ValueType, Functor> function(SystemTag system_tag, int inputs, Functor function=Functor()) {
    return Function<SystemTag, ValueType, Functor>(bc::Dim<1>{inputs}, function);
}

template<class SystemTag, class Functor>
auto function(SystemTag system_tag, int inputs, Functor function=Functor()) {
    return Function<SystemTag, typename SystemTag::default_floating_point_type, Functor>(bc::Dim<1>{inputs}, function);
}


template<class ValueType, class SystemTag, class Functor, int X>
Function<SystemTag, ValueType, Functor> function(SystemTag system_tag, Dim<X> shape, Functor function=Functor()) {
    return Function<SystemTag, ValueType, Functor, bc::traits::Integer<X>>(shape, function);
}

template<class SystemTag, class Functor, int X>
auto function(SystemTag system_tag, bc::Dim<X> shape, Functor function=Functor()) {
    return Function<SystemTag, typename SystemTag::default_floating_point_type, Functor, bc::traits::Integer<X>>(shape, function);
}


}
}

#endif