tensorflow简单使用案例

# fetch:同时运行多个op
import tensorflow as tf

input = tf.constant(3.0)
input2 = tf.constant(2.0)
input3 = tf.constant(3.0)

add = tf.add(input2,input3)
mul = tf.multiply(input1,add)

with tf.Session as sess:
     result = sess.run([mul,add])
     print(result)

 

1.Fetch:执行多个op

2.Feed

#创建占位符

#Feed
#占位符placeholder
input1 = tf.placeholder(tf.float32)
input2 = tf.placeholder(tf.float32)
output = tf.multiply(input1,input2)

with tf.Session() as sess:
    #feed的数据以字典的形式传入
    print(sess.run(output,feed_dict={input1:[8.],input2:[2.]}))

3.简单tensorflow案例

#numpy产生100个随机点
#随机点[0,1)均匀分布
x_data = np.random.rand(100)
y_data = x_data*0.1 + 0.2

#构造一个线性模型
b = tf.Variable(0.)
k = tf.Variable(0.)
y = k*x_data + b

#二次代价函数,square为平方
loss = tf.reduce_mean(tf.square(y_data-y))
#定义一个梯度下降法来进行训练的优化器
optimizer = tf.train.GradientDescentOptimizer(0.2)
#最小化代价函数
train = optimizer.minimize(loss)

#变量初始化
init = tf.global_variables_initializer()

with tf.Session() as sess:
    sess.run(init)
    for step in range(201):
        sess.run(train)
        if step%20==0:
            print(step,sess.run([k,b]))

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是一个使用 TensorFlow C++ API 实现简单的线性回归模型的示例代码: ```cpp #include <iostream> #include "tensorflow/cc/client/client_session.h" #include "tensorflow/cc/ops/standard_ops.h" #include "tensorflow/core/framework/tensor.h" using namespace std; using namespace tensorflow; using namespace tensorflow::ops; int main() { // 构建训练数据 Tensor x(DT_FLOAT, TensorShape({3, 1})); auto x_map = x.tensor<float, 2>(); x_map(0, 0) = 1.0; x_map(1, 0) = 2.0; x_map(2, 0) = 3.0; Tensor y(DT_FLOAT, TensorShape({3, 1})); auto y_map = y.tensor<float, 2>(); y_map(0, 0) = 2.0; y_map(1, 0) = 4.0; y_map(2, 0) = 6.0; // 定义模型 auto scope = Scope::NewRootScope(); auto X = Placeholder(scope, DT_FLOAT); auto Y = Placeholder(scope, DT_FLOAT); auto W = Variable(scope, {1, 1}, DT_FLOAT); auto b = Variable(scope, {1}, DT_FLOAT); auto Y_hat = Add(scope, MatMul(scope, X, W), b); auto loss = Mean(scope, Square(scope, Sub(scope, Y_hat, Y)), {0, 1}); // 构建优化器 auto train_op = GradientDescentOptimizer(scope, 0.01f).Minimize(loss); // 运行会话 ClientSession session(scope); TF_CHECK_OK(session.Run({Assign(scope, W, {{0.0f}}), Assign(scope, b, {0.0f})}, nullptr)); for (int i = 0; i < 100; i++) { TF_CHECK_OK(session.Run({{X, x}, {Y, y}}, {train_op}, nullptr)); } // 打印结果 auto W_tensor = W.eval(session); auto b_tensor = b.eval(session); cout << "W: " << W_tensor.tensor<float, 2>()(0, 0) << endl; cout << "b: " << b_tensor.tensor<float, 1>()(0) << endl; return 0; } ``` 这段代码实现了一个简单的线性回归模型,使用 TensorFlow C++ API 定义了模型、损失函数和优化器,并在训练数据上进行了训练。最后输出训练得到的模型参数。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值