如何用C ++填充张量

Question

我正在创建一个像这样的张量：

tensorflow::Tensor a(tensorflow::DT_FLOAT, tensorflow::TensorShape());

我知道如何填写标量值：

a.scalar<float>()() = 8.0;

但我不知道如何填补像[1,4,2]那样的张量。

Answer 1

有几个选择。 如果张量实际上是一个小向量，就像你的情况一样，你可以做到以下几点：

tensorflow::Tensor a(tensorflow::DT_FLOAT, tensorflow::TensorShape(3));
a.vec<float>()(0) = 1.0f;
a.vec<float>()(1) = 4.0f;
a.vec<float>()(2) = 2.0f;

如果你想构造一个稍大和/或多维的张量，那么在tensorflow/cc/framework/ops.h声明的tensorflow::ops::Input::Initializer有许多构造函数可以让你从各种类型构造一个Tensor C ++常量，例如简单的原始常量和表示多维数组的嵌套初始化列表。

例如，如果要构建2x2矩阵，可以执行以下操作：

#include "tensorflow/cc/framework/cc/ops.h"

tensorflow::ops::Input::Initializer a({{1, 2}, {3, 4}});
// a.tensor will be a Tensor with type DT_INT32 and shape {2, 2}.