检查失败:1 == NumElements() (1 vs. 2)必须有一个元素张量开发者社区

link管理

链接快照平台

输入网页链接，自动生成快照
标签化管理网页链接

文章/答案/技术大牛

发布

tensorflow::Tensor loadImage(tensorflow::string fname){
    tensorflow::int32 width = 224;
    tensorflow::int32 height = 224;
    tensorflow::int32 nData = 1;
    tensorflow::int32 nVec = width*height;
    tensorflow::int32 channels = 3;
    auto tensor = tensorflow::Tensor(tensorflow::DT_FLOAT, tensorflow::TensorShape({1, height, width, channels}));
    auto mat = tensor.tensor<float, 4>();
    std::ifstream fin(fname, std::ios_base::in | std::ios_base::binary);
    assert(!fin.fail());
    boost::iostreams::filtering_istream s;
    s.push(fin);
    char c;
    for(int i=0;i<nData;i++){
        for(int j=0;j<channels;j++){
            for(int h=0;h<height;h++){
                for(int w=0;w<width;w++){
                    s.get(c);
                    mat(i, j, h, w) = static_cast<float>(static_cast<uint8_t>(c)) / 255.0;
    }std::cout << "Image Loaded" << std::endl;
    return tensor;
}

status = session->Run(inputs, {"output_node0"}, {}, &outputs);
auto output_c = outputs[0].scalar<float>(); // an error occurred

auto output_c = outputs[0].scalar<float>();

auto output_c = outputs[0].flat<float>();

error: static assertion failed: YOU_MADE_A_PROGRAMMING_MISTAKE
 #define EIGEN_STATIC_ASSERT(X,MSG) static_assert(X,#MSG);/tmp/tensorflow/include/unsupported/Eigen/CXX11/src/Tensor/TensorMap.h:224:7: note: in expansion of macro 'EIGEN_STATIC_ASSERT'
       EIGEN_STATIC_ASSERT(NumIndices == 0, YOU_MADE_A_PROGRAMMING_MISTAKE)

std::vector<tensorflow::Tensor> outputs;
tensorflow::ClientSession session(root);
TF_CHECK_OK(session.Run({some_operation}, &outputs));
float *result_float_data = outputs[0].flat<float>().data();

tensorflow::Scope root = tensorflow::Scope::NewRootScope();
std::string fn = "Grace_Hopper.png";
auto net1 = tensorflow::ops::ReadFile(root, fn);
auto net2 = tensorflow::ops::DecodePng(root, net1);
auto net3 = tensorflow::ops::Cast(root, net2, tensorflow::DT_FLOAT);
auto net4 = tensorflow::ops::ExpandDims(root, net3, 0);
// do something here
auto net5 = tensorflow::ops::ResizeBilinear(root, net4, tensorflow::ops::Const(root, {2 * 606, 2 * 517}));
auto net6 = tensorflow::ops::Reshape(root, net5, tensorflow::ops::Const(root, {2 * 606, 2 * 517, 3}));
// convert back
auto net7 = tensorflow::ops::Cast(root, net6, tensorflow::DT_UINT8);
auto net8 = tensorflow::ops::EncodeJpeg(root, net7);

问 检查失败:1 == NumElements() (1 vs. 2)必须有一个元素张量 EN

回答 1

Stack Overflow用户