TensorFlow

A simple code example

import numpy as np 

import tensorflow as tf

## Convert to version 1.0

#import tensorflow.compat.v1 as tf

#tf.disable_v2_behavior()



coefficients = np.array([[1], [-16], [64]])

w = tf.Variable([0], dtype=tf.float32)

x = tf.placeholder(tf.float32, [3,1])

cost = x[0][0]*w**2 + x[1][0]*w + x[2][0]     #(w-8)**2

# Or, tf.matmul(a,b) to do a matrix multiplication; tf.add(a,b) to do an addition

train = tf.train.GradientDescentOptimizer(0.01).minimize(cost)

init = tf.global_variables_initializer()



session = tf.Session()

session.run(init)

print(session.run(w))

###################

## The with command in Python is a little bit better at cleaning up in cases an error in exception while executing this inner loop.

## And this takes care of closing the session for you

# with tf.Session() as session:

#   session.run(init)

#   print(session.run(w))

#############



for i in range(1000):

  session.run(train, feed_dict = {x:coefficients})

print(session.run(w))

session.close()    # You don't need this if you use the with command

Output:

[0.]
[7.9999886]

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值