Method GetOutputSampleShape

src/model/layer/split.h:35–38

  /// \copydoc Layer::Setup(const LayerConf&);
  void Setup(const Shape& in_sample, const LayerConf& conf) override;
  const Shape GetOutputSampleShape() const override {
    CHECK(out_sample_shape_.size()) << "You may haven't call Setup()";
    return out_sample_shape_;
  }
  /// The inputs should have only one Tensor
  /// The outputs is a set of replicated Tensor
  const vector<Tensor> Forward(int flag, const vector<Tensor> &inputs) override;