Method Setup

src/model/layer/split.cc:28–33

RegisterLayerClass(singacl_split, Split);

void Split::Setup(const Shape& in_sample, const LayerConf& conf) {
  Layer::Setup(in_sample, conf);
  SplitConf split_conf = conf.split_conf();
  output_size_ = split_conf.output_size();
  out_sample_shape_ = in_sample;
}

const vector<Tensor> Split::Forward(int flag, const vector<Tensor>& inputs) {
  vector<Tensor> outputs;