代码拉取完成,页面将自动刷新
// Tencent is pleased to support the open source community by making ncnn available.
//
// Copyright (C) 2020 THL A29 Limited, a Tencent company. All rights reserved.
//
// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
// in compliance with the License. You may obtain a copy of the License at
//
// https://opensource.org/licenses/BSD-3-Clause
//
// Unless required by applicable law or agreed to in writing, software distributed
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.
#include "testutil.h"
static int test_concat(const std::vector<ncnn::Mat>& a, int axis)
{
ncnn::ParamDict pd;
pd.set(0, axis); //axis
std::vector<ncnn::Mat> weights(0);
int ret = test_layer("Concat", pd, weights, a);
if (ret != 0)
{
fprintf(stderr, "test_concat failed a[0].dims=%d a[0]=(%d %d %d %d) axis=%d\n", a[0].dims, a[0].w, a[0].h, a[0].d, a[0].c, axis);
}
return ret;
}
static int test_concat_0()
{
ncnn::Mat a[] = {
RandomMat(15, 5, 6, 13),
RandomMat(15, 5, 6, 20),
RandomMat(15, 5, 6, 24),
RandomMat(15, 5, 6, 48)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
for (int j = 0; j < n; j++)
{
for (int k = 0; k < n; k++)
{
std::vector<ncnn::Mat> as(4);
as[0] = a[i];
as[1] = a[j];
as[2] = a[k];
as[3] = a[k];
int ret = test_concat(as, 0) || test_concat(as, -4);
if (ret != 0)
return ret;
}
}
}
return 0;
}
static int test_concat_1()
{
ncnn::Mat a[] = {
RandomMat(15, 3, 15, 13),
RandomMat(15, 3, 16, 20),
RandomMat(15, 3, 17, 24),
RandomMat(15, 3, 18, 48)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
std::vector<ncnn::Mat> as(3);
as[0] = a[i];
as[1] = a[i];
as[2] = a[i];
int ret = test_concat(as, 1) || test_concat(as, -3);
if (ret != 0)
return ret;
}
return 0;
}
static int test_concat_2()
{
ncnn::Mat a[] = {
RandomMat(15, 15, 6, 13),
RandomMat(15, 16, 6, 20),
RandomMat(15, 17, 6, 24),
RandomMat(15, 18, 6, 48)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
std::vector<ncnn::Mat> as(3);
as[0] = a[i];
as[1] = a[i];
as[2] = a[i];
int ret = test_concat(as, 2) || test_concat(as, -2);
if (ret != 0)
return ret;
}
return 0;
}
static int test_concat_3()
{
ncnn::Mat a[] = {
RandomMat(15, 5, 7, 13),
RandomMat(16, 5, 7, 20),
RandomMat(17, 5, 7, 24),
RandomMat(18, 5, 7, 48)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
std::vector<ncnn::Mat> as(3);
as[0] = a[i];
as[1] = a[i];
as[2] = a[i];
int ret = test_concat(as, 3) || test_concat(as, -1);
if (ret != 0)
return ret;
}
return 0;
}
static int test_concat_4()
{
ncnn::Mat a[] = {
RandomMat(15, 13, 13),
RandomMat(15, 13, 20),
RandomMat(15, 13, 24),
RandomMat(15, 13, 48)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
for (int j = 0; j < n; j++)
{
for (int k = 0; k < n; k++)
{
std::vector<ncnn::Mat> as(4);
as[0] = a[i];
as[1] = a[j];
as[2] = a[k];
as[3] = a[k];
int ret = test_concat(as, 0) || test_concat(as, -3);
if (ret != 0)
return ret;
}
}
}
return 0;
}
static int test_concat_5()
{
ncnn::Mat a[] = {
RandomMat(15, 15, 13),
RandomMat(15, 16, 20),
RandomMat(15, 17, 24),
RandomMat(15, 18, 48)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
std::vector<ncnn::Mat> as(3);
as[0] = a[i];
as[1] = a[i];
as[2] = a[i];
int ret = test_concat(as, 1) || test_concat(as, -2);
if (ret != 0)
return ret;
}
return 0;
}
static int test_concat_6()
{
ncnn::Mat a[] = {
RandomMat(15, 13, 13),
RandomMat(16, 13, 20),
RandomMat(17, 13, 24),
RandomMat(18, 13, 48)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
std::vector<ncnn::Mat> as(3);
as[0] = a[i];
as[1] = a[i];
as[2] = a[i];
int ret = test_concat(as, 2) || test_concat(as, -1);
if (ret != 0)
return ret;
}
return 0;
}
static int test_concat_7()
{
ncnn::Mat a[] = {
RandomMat(19, 29),
RandomMat(19, 44),
RandomMat(19, 56),
RandomMat(19, 80)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
for (int j = 0; j < n; j++)
{
for (int k = 0; k < n; k++)
{
std::vector<ncnn::Mat> as(4);
as[0] = a[i];
as[1] = a[j];
as[2] = a[k];
as[3] = a[k];
int ret = test_concat(as, 0) || test_concat(as, -2);
if (ret != 0)
return ret;
}
}
}
return 0;
}
static int test_concat_8()
{
ncnn::Mat a[] = {
RandomMat(19, 29),
RandomMat(16, 44),
RandomMat(17, 56),
RandomMat(18, 80)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
std::vector<ncnn::Mat> as(3);
as[0] = a[i];
as[1] = a[i];
as[2] = a[i];
int ret = test_concat(as, 1) || test_concat(as, -1);
if (ret != 0)
return ret;
}
return 0;
}
static int test_concat_9()
{
ncnn::Mat a[] = {
RandomMat(29),
RandomMat(44),
RandomMat(56),
RandomMat(80)
};
const int n = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < n; i++)
{
for (int j = 0; j < n; j++)
{
for (int k = 0; k < n; k++)
{
std::vector<ncnn::Mat> as(4);
as[0] = a[i];
as[1] = a[j];
as[2] = a[k];
as[3] = a[k];
int ret = test_concat(as, 0) || test_concat(as, -1);
if (ret != 0)
return ret;
}
}
}
return 0;
}
int main()
{
SRAND(7767517);
return 0
|| test_concat_0()
|| test_concat_1()
|| test_concat_2()
|| test_concat_3()
|| test_concat_4()
|| test_concat_5()
|| test_concat_6()
|| test_concat_7()
|| test_concat_8()
|| test_concat_9();
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。