将2d std::数组传递给函数cpp

Passing 2d std::array to function cpp

本文关键字:函数 cpp 数组 2d std      更新时间:2023-10-16

我试图在c++中编写一个函数,该函数将接受2个输入std::数组,并通过矩阵乘法返回产品数组。然而,该函数不能采用不同尺寸的数组(例如4x4工作,3x4不)

下面是代码:

#include <iostream> 
#include <array>

template <std::size_t SIZE>
void dot(std::array<std::array<double, SIZE>, SIZE>& array1, 
     std::array<std::array<double, SIZE>, SIZE>& array2)
{
  int x1 = array1.size();
  int y1 = array1[0].size();

  int x2 = array2.size();
  int y2 = array2[0].size();
}
int main()
{
  using std::array;
  array<array<double, 4>, 4> syn0 = {{ {1,2,4},
                       {2,3,4},
                       {6,8,6},
                       {1,2,4} }};
  dot(syn0, syn0);
  return 0;
}

使用这个问题中的模板示例,它将接受像代码中的4x4这样的数组。

将矩阵更改为不相等的数字会产生以下错误:

newer.cpp: In function ‘int main()’:
newer.cpp:23:21: error: too many initializers for ‘std::__array_traits<std::array<double, 4ul>, 3ul>::_Type {aka std::array<double, 4ul> [3]}’
            {1,2,4} }};
                     ^
newer.cpp:24:17: error: no matching function for call to ‘dot(std::array<std::array<double, 4ul>, 3ul>&, std::array<std::array<double, 4ul>, 3ul>&)’
   dot(syn0, syn0);
                 ^
newer.cpp:24:17: note: candidate is:
newer.cpp:6:6: note: template<long unsigned int SIZE> void dot(std::array<std::array<double, SIZE>, SIZE>&, std::array<std::array<double, SIZE>, SIZE>&)
 void dot(std::array<std::array<double, SIZE>, SIZE>& array1, 
      ^
newer.cpp:6:6: note:   template argument deduction/substitution failed:
newer.cpp:24:17: note:   deduced conflicting values for non-type parameter ‘SIZE’ (‘4ul’ and ‘3ul’)
   dot(syn0, syn0);
                 ^
newer.cpp:24:17: note:   ‘std::array<std::array<double, 4ul>, 3ul>’ is not derived from ‘std::array<std::array<double, SIZE>, SIZE>

我认为这样做的原因是模板只分配了一个变量,所以如果我将2赋值给同一个变量,它就会抛出错误。我测试了是否可以为两个不同的变量堆叠模板,但这是不可能的。

我怎么能允许函数采取任何大小的二维数组,而不导致错误?

看来你们真的很接近了。

你只需要把SIZE2作为一个参数添加到你的模板中:

template <std::size_t SIZE,std::size_t SIZE2>
void dot(std::array<std::array<double, SIZE>, SIZE2>& array1, 
     std::array<std::array<double, SIZE>, SIZE2>& array2)

,它可以正常编译顺便说一句,你的syn0尺寸错了

int main()
{
  using std::array;
  array<array<double, 3>, 4> syn0 = {{ {1,2,4},
                       {2,3,4},
                       {6,8,6},
                       {1,2,4} }};
  dot(syn0, syn0);
  return 0;
}