c++ - 如何在不指定维度的情况下使用 Eigen 声明张量?
问题描述
我有一个函数将 n 维张量作为输入,我必须存储这个张量以便以后重用它。但是,我事先不知道我的张量的维度。我想这样做:
//in class.h
Eigen::Tensor<double, N> mytensor;
//in class.cpp
mytensor = input;
有没有办法做到这一点?
解决方案
由于参数N
是非类型模板参数,它必须是编译时已知的值。这意味着您不能真正将Eigen::Tensor
with unspecified存储N
在变量中,因为每个具有不同大小的实例都是不同的类型。
您可以通过使用容器(例如std::variant
和)来解决此问题std::any
。它们可以分别存储来自封闭类型集或任何类型的类型的对象。例如,您可以使用std::any
创建tensor_holder
如下的辅助类:
template <typename T>
class tensor_holder
{
private:
std::any _held;
size_t _size;
public:
template <int N>
constexpr tensor_holder(Eigen::Tensor<T, N> tensor) :
_held{std::move(tensor)},
_size{N}
{
}
constexpr tensor_holder(const tensor_holder&) = default;
constexpr tensor_holder(tensor_holder&&) = default;
template <size_t N>
Eigen::Tensor<T, N>& get()
{
return std::any_cast<Eigen::Tensor<T, N>&>(_held);
}
template <size_t N>
const Eigen::Tensor<T, N>& get() const
{
return std::any_cast<Eigen::Tensor<T, N>&>(_held);
}
constexpr int size() const noexcept
{
return _size;
}
};
它将Eigen::Tensor
任意大小的 a 存储在std::any
. 如果提供的大小与实际张量大小不匹配,则成员函数get
将抛出std::bad_any_cast
(请注意,大小作为编译时间值的限制不会消失)。
假设您有一小组可能的张量大小,您可以在函数中运行不同的分支,如下所示:
void use_tensor(const tensor_holder<double>& in)
{
static tensor_holder held = in; // it can be stored for later
if (held.size() == 4)
{
auto& tensor = held.get<4>();
tensor(0, 1, 2, 3) = 115.5;
}
else if (held.size() == 3)
{
// some other logic
}
}
在这种情况下,您也可以使用 aswitch
而不是a。if
仅当您必须支持的不同张量大小的数量很少时,此解决方案才实用。我不知道你打算如何使用你的张量,所以我很难说这个解决方案是否足够。
由于 class 的转换构造函数,您可以通过简单地提供任意大小tensor_holder
的函数来调用该函数:Eigen::Tensor
use_tensor(Eigen::Tensor<double, 3>{});
(神螺栓)
推荐阅读
- javascript - 我如何在这个应用程序中使用 jest?
- arduino - 从 DeepSleep 模式返回后 Mpu6050 未连接到 Esp32
- r - 是否有一种功能(或一种优化方法)可以比循环更快地读取许多文本文件?
- svelte-3 - 最初显示带有表单的 Svelte 页面时,文本框始终“自动填充”
- angular - 有没有办法从谷歌分析中创建角度报告数据?
- python - 带索引变量的符号向量范数
- c++ - 通过两种方式声明二维向量的区别
- python - 带有拟合 Von Mises 分布的圆形直方图
- apache-spark - 在 S3 中捕获 Kubernetes Spark 驱动程序和执行程序日志并在历史服务器中查看
- postgresql - PostgreSQL DO WHILE 循环