fix weights rank assertion in InnerProductOp

2020-02-22 16:59:09 +05:30
parent 96b26dc8f4
commit c23ab37355
1 changed files with 1 additions and 1 deletions
@@ -31,7 +31,7 @@ namespace cv { namespace dnn { namespace cuda4dnn {
            : stream(std::move(stream_)), cublasHandle(std::move(handle)), axis{ axis }
        {
            weightsTensor = csl::makeTensorHeader<T>(weights);
-            CV_Assert(get_effective_rank(weightsTensor) == 2);
+            CV_Assert(get_effective_rank(weightsTensor) <= 2);
            csl::copyMatToTensor<T>(weights, weightsTensor, stream);

            if (!bias.empty())