28#include "tiny_dnn/layers/layer.h"
35 :
layer({vector_type::data}, {vector_type::data}), shape_(shape) {}
40 std::vector<shape3d>
in_shape()
const override {
return { shape_ }; }
41 std::vector<shape3d>
out_shape()
const override {
return { shape_ }; }
42 std::string
layer_type()
const override {
return "input"; }
47 std::vector<tensor_t*>&
out_data)
override {
52 const std::vector<tensor_t*>&
out_data,
54 std::vector<tensor_t*>&
in_grad)
override {
56 CNN_UNREFERENCED_PARAMETER(
in_data);
57 CNN_UNREFERENCED_PARAMETER(
out_data);
58 CNN_UNREFERENCED_PARAMETER(
out_grad);
59 CNN_UNREFERENCED_PARAMETER(
in_grad);
Simple image utility class.
Definition image.h:94
base class of all kind of NN layers
Definition layer.h:62