CaffeOnACL/include/caffe/layers/softmax_layer.hpp at master · dlyshare/CaffeOnACL · GitHub

Repository navigation

History

50 lines (41 loc) · 1.53 KB

Raw

#ifndef CAFFE_SOFTMAX_LAYER_HPP_

#define CAFFE_SOFTMAX_LAYER_HPP_

#include <vector>

#include "caffe/blob.hpp"

#include "caffe/layer.hpp"

#include "caffe/proto/caffe.pb.h"

namespace caffe {

/**

* @brief Computes the softmax function.

* TODO(dox): thorough documentation for Forward, Backward, and proto params.

template <typename Dtype>

class SoftmaxLayer : public Layer<Dtype> {

public:

explicit SoftmaxLayer(const LayerParameter& param)

: Layer<Dtype>(param) {}

virtual void Reshape(const vector<Blob<Dtype>*>& bottom,

const vector<Blob<Dtype>*>& top);

virtual inline const char* type() const { return "Softmax"; }

virtual inline int ExactNumBottomBlobs() const { return 1; }

virtual inline int ExactNumTopBlobs() const { return 1; }

protected:

virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,

const vector<Blob<Dtype>*>& top);

virtual void Forward_gpu(const vector<Blob<Dtype>*>& bottom,

const vector<Blob<Dtype>*>& top);

virtual void Backward_cpu(const vector<Blob<Dtype>*>& top,

const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom);

virtual void Backward_gpu(const vector<Blob<Dtype>*>& top,

const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom);

int outer_num_;

int inner_num_;

int softmax_axis_;

/// sum_multiplier is used to carry out sum using BLAS

Blob<Dtype> sum_multiplier_;

/// scale is an intermediate Blob to hold temporary results.

Blob<Dtype> scale_;

};

} // namespace caffe

#endif // CAFFE_SOFTMAX_LAYER_HPP_

CaffeOnACL/include/caffe/layers/softmax_layer.hpp at master · dlyshare/CaffeOnACL · GitHub

Repository navigation

Expand file tree

Breadcrumbs

softmax_layer.hpp

Latest commit

History

Breadcrumbs

softmax_layer.hpp

File metadata and controls