diff --git a/src/layer/concat.cpp b/src/layer/concat.cpp index 6c1b9dc89..72b8bb3a5 100644 --- a/src/layer/concat.cpp +++ b/src/layer/concat.cpp @@ -24,6 +24,43 @@ Concat::Concat() int Concat::forward(const std::vector& bottom_blobs, std::vector& top_blobs) const { + int dims = bottom_blobs[0].dims; + + if (dims == 1) + { + // concat vector + // total length + int top_w = 0; + for (size_t b=0; b