小编Ros*_*oss的帖子

如何设置 3D tensorflow Tensor c++ 的元素

我有以下 3D 张量:

auto inputX = Tensor(DT_DOUBLE, TensorShape({1,1,9}));
Run Code Online (Sandbox Code Playgroud)

如何为张量中的元素设置值?

另外对于下面的一维张量,我将如何设置一个值?

auto inputY = Tensor(DT_INT32, TensorShape({1}));
Run Code Online (Sandbox Code Playgroud)

谢谢!

更新如下:

这是我的代码:

#include <iostream>
#include <stdlib.h>
#include <fstream>
#include <string.h>
#include <vector>
#include <sstream>
#include <typeinfo>
#include <sys/time.h>
#include "tensorflow/cc/client/client_session.h"
#include "tensorflow/cc/ops/standard_ops.h"
#include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/public/session.h"
#include "tensorflow/core/platform/env.h"

using namespace std;
using namespace tensorflow;
using namespace tensorflow::ops;

Session* session; // tensorflow session
Status status;  // tensorflow status

double get_prediction(Tensor a);

double get_prediction(Tensor a, Tensor b) {

  double prediction;
  vector<Tensor> outputs;

  // preparing input
  // …
Run Code Online (Sandbox Code Playgroud)

c++ tensorflow

3
推荐指数
1
解决办法
1121
查看次数

标签 统计

c++ ×1

tensorflow ×1