#include <tensorflow/cc/client/client_session.h>
#include <tensorflow/cc/ops/standard_ops.h>
#include <vector>

using namespace tensorflow;
using namespace tensorflow::ops;

int main() {
  // 创建会话
  ClientSession session(Env::Default());

  // 加载数据
  std::vector<float> features = {1.0, 2.0, 3.0, 4.0};
  std::vector<float> labels = {1.0, 2.0, 3.0, 4.0};

  // 创建占位符
  Placeholder features_placeholder(DT_FLOAT, TensorShape({4, 1}));
  Placeholder labels_placeholder(DT_FLOAT, TensorShape({4, 1}));

  // 创建变量
  Variable weights(DT_FLOAT, TensorShape({1, 1}), "weights");

  // 构建图
  MatMul matmul = MatMul(features_placeholder, weights);
  BiasAdd bias_add = BiasAdd(matmul);

  // 创建损失函数
  MeanSquaredError loss = MeanSquaredError(bias_add, labels_placeholder);

  // 创建优化器
  GradientDescentOptimizer optimizer(0.1);

  // 优化过程
  TrainOp train_op = optimizer.Minimize(loss);

  // 初始化变量
  session.Run(InitializeAllVariablesOp());

  // 训练神经网络
  for (int i = 0; i < 1000;   i) {
    // 构建输入数据
    Tensor features_tensor(DT_FLOAT, TensorShape({4, 1}), features);
    Tensor labels_tensor(DT_FLOAT, TensorShape({4, 1}), labels);

    // 训练一步
    session.Run(train_op, {{"features", features_tensor}, {"labels", labels_tensor}});
  }

  // 获取权重
  Tensor weights_tensor;
  session.Run(weights.ReadValue(&weights_tensor));

  // 输出权重
  std::cout << "权重：" << weights_tensor.DebugString() << std::endl;

  return 0;
}