forked from btgraham/SparseConvNet-archived
-
Notifications
You must be signed in to change notification settings - Fork 0
/
TerminalPoolingLayer.h
24 lines (23 loc) · 931 Bytes
/
TerminalPoolingLayer.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#pragma once
#include <iostream>
#include "SpatiallySparseLayer.h"
#include "Rng.h"
class TerminalPoolingLayer : public SpatiallySparseLayer {
public:
int inSpatialSize; //==poolSize.
int outSpatialSize; // 1
int poolSize;
int S; // Maximum number of active sites
TerminalPoolingLayer(cudaMemStream &memStream, int poolSize, int S);
void preprocess(SpatiallySparseBatch &batch,
SpatiallySparseBatchInterface &input,
SpatiallySparseBatchInterface &output);
void forwards(SpatiallySparseBatch &batch,
SpatiallySparseBatchInterface &input,
SpatiallySparseBatchInterface &output);
void backwards(SpatiallySparseBatch &batch,
SpatiallySparseBatchInterface &input,
SpatiallySparseBatchInterface &output, float learningRate,
float momentum);
int calculateInputSpatialSize(int outputSpatialSize);
};