带有 std::variant 和 std::visit 的联合模板

Question

我正在使用一个模板库，因为在我的代码中，一个模板参数可以假设一个有限范围的值，我决定在建议下使用 std::variant 并在其中声明我可能会使用的所有对象需要：

std::variant<TemplateClass<1>, TemplateClass<2>, ..., TemplateClass<5>>

我从来没有用过这个工具。

要访问 TemplateClass 的方法，我必须使用 std::visit，但有时它可以工作，而其他情况则不能，比如 no member function XXX in std::variant < .... > 或“在函数模板专业化的实例化中......” [我什至没有了解这里有什么问题]

具体来说，我正在使用Eigen::Tensor 库，当我调用rank()、dimension(n) 之类的方法时，它可以工作，而对于dimensions() 和setRandom() 之类的方法则不起作用。

下面是我的实施草稿

std::variant<Eigen::Tensor<double, 1>, Eigen::Tensor<double, 2>, /* ... */> makeTensor(
    int i, const std::initializer_list<int> dims) {
  switch (i) {
    case 1: {
      Eigen::Tensor<double, 1> T1;
      T1.resize(dims);
      return T1;
    }

    case 2: {
      Eigen::Tensor<double, 2> T2;
      T2.resize(dims);
      return T2;
    }

      /* ... */
  }
}

int main() {
  auto myTensor{makeTensor(2, {4, 5})};  // Tensor 2D 4x5

  // Working methods
  auto rnk = std::visit([](const auto &tensor) { return tensor.rank(); }, myTensor);

  auto dim1 = std::visit([](const auto &tensor) { return tensor.dimension(0); }, myTensor);

  // Not working methods

  auto dimsTens =
      std::visit([](const auto &tensor) { return tensor.dimensions(); }, myTensor);  // 5 times same error saying
  //'In instantiation of function template specialization 'std::visit<(lambda at
  /// home/virginie/Desktop/Project/main.cpp:62:33),
  // std::variant<Eigen::Tensor<double, 1, 0, long>, Eigen::Tensor<double, 2, 0, long>, Eigen::Tensor<double, 3, 0,
  // long>, Eigen::Tensor<double, 4, 0, long>, Eigen::Tensor<double, 5, 0, long>> &>''

  std::visit([&myTensor]() { myTensor.setRandom(); });  // 'No member setRandom() in std::variant<...>'
}

我是否以错误的方式使用std::visit？

---- 编辑----

在@florestan 的建议下，我已经解决了与dimensions() 相关的问题，而setRandom 我得到了以下信息：

在 /..../ main.cpp 包含的文件中 从这里需要

Answer 1

您需要为所有可能的选择返回相同的类型。以dimension为例，需要将数组元素复制到向量中。

这样的事情应该会有所帮助：

 auto dimsTens=std::visit(
    [](const auto &tensor) {
       auto dims = tensor.dimensions();
       return std::vector<int>(dims.begin(), dims.end());
    }, myTensor);

第二个错误是因为您没有以正确的方式调用std::visit。它需要两个参数，第一个是要访问的函数，第二个是要访问的变量。以下应该可以工作。

std::visit([](auto& t){ t.setRandom();}, myTensor); 

实时代码： https://godbolt.org/z/vq4PYo