shin/NeuralNetworkLib
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

new function to support LSTM Unit

Browse Source
This commit is contained in:
Shin
2016-01-27 23:40:32 +01:00
parent d424d87535
commit 3c26c9641c
7 changed files with 97 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/NeuralNetwork/BasisFunction/Linear.h
View File

@@ -17,7 +17,7 @@ namespace BasisFunction {
public:
Linear() {}
inline virtual float computeStreaming(const std::vector<float>& weights, const std::vector<float>& input) override {
inline virtual float computeStreaming(const std::vector<float>& weights, const std::vector<float>& input) const override {
size_t inputSize=input.size();
size_t alignedPrev=inputSize-inputSize%4;
@@ -46,7 +46,7 @@ namespace BasisFunction {
return partialSolution.f[0];
}
inline virtual float compute(const std::vector<float>& weights, const std::vector<float>& input) override {
inline virtual float compute(const std::vector<float>& weights, const std::vector<float>& input) const override {
register float tmp = 0;
size_t inputSize=input.size();
for(size_t k=0;k<inputSize;k++) {