outputlayer.h

Go to the documentation of this file.

#include <vector>
#include "layer/layer.h"
#include "tensor/tensor.h"
#include "compute/operation.h"
#include "compute/tensor_operations.h"

namespace magmadnn {
namespace layer {

template <typename T>
class OutputLayer : public Layer<T> {
public:
    OutputLayer(op::Operation<T> *input);
    virtual ~OutputLayer();

    virtual std::vector<op::Operation<T> *> get_weights();

protected:
    void init();

};

template <typename T>
OutputLayer<T>* output(op::Operation<T> *input);

}   // layer
}   // magmadnn