bert/research-project
Archived
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Use caffe multiplication for efficiency.

Browse Source
This commit is contained in:
Bert Peters
2018-02-23 16:05:34 +01:00
parent a8857faa24
commit 63b671f54d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/visualisations.cpp
View File

@@ -1,5 +1,6 @@
#include <algorithm> #include <algorithm>
#include <numeric> #include <numeric>
#include <util/math_functions.hpp>
#include "visualisations.hpp" #include "visualisations.hpp"
#include "DummyLayerVisualisation.hpp" #include "DummyLayerVisualisation.hpp"
#include "MultiImageVisualisation.hpp" #include "MultiImageVisualisation.hpp"
@@ -92,9 +93,10 @@ static Animation *getFullyConnectedAnimation(const fmri::LayerData &prevState, c
const auto numEntries = accumulate(shape.begin(), shape.end(), static_cast<size_t>(1), multiplies<void>()); const auto numEntries = accumulate(shape.begin(), shape.end(), static_cast<size_t>(1), multiplies<void>());
vector<float> interactions(numEntries); vector<float> interactions(numEntries);
const auto stepSize = shape[0];
for (auto i : Range(numEntries)) { for (auto i : Range(numEntries / stepSize)) {
interactions[i] = weights[i] * data[i % shape[0]]; caffe::caffe_mul(shape[0], &weights[i * stepSize], data, &interactions[i * stepSize]);
} }
const auto desiredSize = min(INTERACTION_LIMIT, numEntries); const auto desiredSize = min(INTERACTION_LIMIT, numEntries);