Function TEST

test/singa/test_concat.cc:27–37

using singa::Shape;

TEST(Concat, Setup) {
  Shape s1{2u, 3u};
  Shape s2{1u, 3u};
  singa::LayerConf conf;
  conf.set_type("singa_concat");
  conf.mutable_concat_conf()->set_axis(0);
  singa::Concat layer;
  layer.Setup({{3u}, {3u}}, conf);
  auto s = layer.GetOutputSampleShape();
  EXPECT_EQ(s[0], 3u);
}

void ForwardConcatRowTest(std::shared_ptr<singa::Device> dev) {
  size_t a = 2u, b = 1u, c = 3u;