rec/fdeep/layers/linear_layer.hpp
2020-03-18 14:42:46 +08:00

31 lines
628 B
C++
Executable File

// Copyright 2016, Tobias Hermann.
// https://github.com/Dobiasd/frugally-deep
// Distributed under the MIT License.
// (See accompanying LICENSE file or at
// https://opensource.org/licenses/MIT)
#pragma once
#include "fdeep/layers/activation_layer.hpp"
#include <string>
namespace fdeep { namespace internal
{
class linear_layer : public activation_layer
{
public:
explicit linear_layer(const std::string& name)
: activation_layer(name)
{
}
protected:
tensor5 transform_input(const tensor5& in_vol) const override
{
return in_vol;
}
};
} } // namespace fdeep, namespace internal