tensorflow - 在 C++ 中 reshape 张量

Question

TensorFlow 的 C++ 接口(interface)似乎没有 reshape 方法。有谁知道如何转换例如[A,B,C,D]进入 [A*B,C,D] ?看起来这样做的唯一方法是使用 Eigen？但是，那里的文档非常薄，代码是模板 hell ，不容易解析。

Answer 1

检查重构张量是否具有与源张量相同数量的元素的解决方案:

// Extracted image features from MobileNet_224
tensorflow::Tensor image_features(tensorflow::DT_FLOAT,
                                  tensorflow::TensorShape({1, 14, 14, 512}));

tensorflow::Tensor image_features_reshaped(tensorflow::DT_FLOAT,
                                           tensorflow::TensorShape({1, 196, 512}));

// Reshape tensor from [1, 14, 14, 512] to [1, 196, 512]
if(!image_features_reshaped.CopyFrom(image_features, tensorflow::TensorShape({1, 196, 512})))
{
  LOG(ERROR) << "Unsuccessfully reshaped image features tensor [" << image_features.DebugString() << "] to [1, 196, 512]";
  return false;
}

LOG(INFO) << "Reshaped features tensor: " << image_features_reshaped.DebugString();