【Tensorflow】op的理解和自定义损失函数

tensorflow中的基本概念

本文是在阅读官方文档后的一些个人理解。

官方文档地址:https://www.tensorflow.org/versions/r0.12/get_started/basic_usage.html#basic-usage

关于tensor和op的理解

Nodes in the graph are called ops (short for operations). 
An op takes zero or more Tensors, performs some computation, and produces zero or more Tensors.

类比:一个神经元有多个输入,一个或者多个输出。这里的OP可以看作神经元,tensor可以看作输入的数据。

In TensorFlow terminology, a Tensor is a typed multi-dimensional array.
For example, you can represent a mini-batch of images as a 4-D array of floating point numbers with dimensions [batch, height, width,channels].

tensor是一个数组,每个数组元素是多维的,其实就是一个矩阵。

TensorFlow程序的阶段

TensorFlow程序通常被组织成一个构建阶段和一个执行阶段在构建阶段,op 的执行步骤被描述成一个图。在执行阶段,使用会话执行执行图中的op。

阶段一:如何构建图?

1.构建图从创建op开始。有些op的创建是不需要input的,比如Constant。这样的op被成为源op(source op)。
2.在python中op对象是由op构造器(ops constructors)创建的。op构造器创建一个op对象时可以传递一个源op作为待构造op对象的输入。
3.op对象被op构造器创建后是作为一个node加入到graph中的。TensorFlow Python 库有一个默认图 (default graph), op 构造器可以为其增加节点。这个默认图对许多程序来说已经足够用了。

总结:因为graph是由op对象组成的,所以构建图的过程其实就是创建op对象的过程,以及如果将这些个op对象连接起来(比如某个op对象作为另外某个op对象的输入)的过程

阶段二:图构建好了,如何执行?

1.因为graph需要在session中启动。所以为了启动一个graph,第一步就是创建session对象。
2.sessoin对象创建的时候如果不制定graph,则使用默认图(default graph)

对Variable的理解

变量用于维护图执行过程中的状态信息。

通常会将一个统计模型中的参数表示为一组变量。 例如, 你可以将一个神经网络的权重作为一个tensor存储在某个变量中。在训练过程中, 通过重复运行训练图,更新这个 tensor。

对feeds和fetches的理解

可以为任意的op赋值或者从其中获取数据。

对于tensor、Constants、Variables三者的理解

tensor存储在Constants或者Variables。就像数据可以放在常量和变量中一样。放在变量中的数据是可以修改的,放在常量中的数据是不可以修改的。

常量op也算是op吧,只是比较简单而已。

原地址:https://www.cnblogs.com/tsiangleo/p/6145112.html


自定义损失函数

import tensorflow as tf
from numpy.random import RandomState
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

batch_size = 8

#两个输入节点。
x = tf.placeholder(tf.float32,shape=(None,2),name='x-input')
#一个输出节点。
y_ = tf.placeholder(tf.float32,shape=(None),name='y-input')

#定义了一个单层的神经网络前向传播过程,这里就是简单加权和。
w1 = tf.Variable(tf.random_normal([2,1],stddev=1,seed=1))
y = tf.matmul(x,w1)

#自定义损失函数
#定义预测多了和预测少了的成本。
loss_less = 10
loss_more = 1
#此处用到了tf.greater 和 tf.where 来实现选择操作,tf.greater 的输入是两个张量,此函数会比较这两个输入张量中每个元素的大小,并返回比较结果。
#当tf.greater的输入张量维度不一样时,TensorFlow会进行类似NumPy广播操作处理。
#tf.where函数有三个参数,第一个为选择条件根据,当为True时,tf.where函数选择第二个参数中的值,否则使用第三个参数中的值。
loss = tf.reduce_sum(tf.where(tf.greater(y,y_),(y - y_)*loss_more,(y_ - y)*loss_less))
train_step = tf.train.AdamOptimizer(0.001).minimize(loss)

#通过随机数生成一个模拟数据集
rdm = RandomState(1)
dataset_size = 128
X = rdm.rand(dataset_size,2)
#设置回归的正确值为两个输入的和加上一个随机变量。之所以要加上一个随机量是为了加入不可预测的噪音,
#否则不同损失函数的意义就不大了,因为不同损失函数都会在能完全预测正确的时候最低。一般来说噪音为一个均值为0的小量,
#所以这里的噪音设置为-0.05~0.05的随机数。
Y = [[x1 + x2 +rdm.rand()/10.0-0.05] for (x1,x2) in X]

#训练神经网络。
with tf.Session() as sess:
    init_op = tf.global_variables_initializer()
    sess.run(init_op)
    STEPS = 5000
    for i in range(STEPS):
        start = (i * batch_size) % dataset_size
        end = min(start+batch_size,dataset_size)
        sess.run(train_step,
                 feed_dict={x:X[start:end],y_:Y[start:end]})
    print (sess.run(w1))

'''
输出结果:
[[1.0193471]
 [1.0428091]]
'''

 

  • 1
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
TensorFlow中的op,即操作对象,是TensorFlow图中的节点,用于执行某种特定的计算或操作。每个op可以接收零个或多个输入Tensor,并且可以输出零个或多个Tensor。op通过操作函数(如tf.matmul())创建。 在TensorFlow中,可以使用自定义op来扩展TensorFlow的功能。自定义op可以用C++或Python编写,并通过注册到TensorFlow中使其可用。使用C++编写自定义op需要定义op的输入、输出和计算逻辑,然后将其编译为TensorFlow的插件。使用Python编写自定义op则更加简单,只需要定义op的计算逻辑即可。 要在TensorFlow中使用自定义op,需要先注册op并指定其输入输出的数据类型和形状。注册op可以使用REGISTER_OP宏,并通过SetShapeFn函数指定op的形状推断规则。形状推断规则可以根据输入Tensor的形状推断出输出Tensor的形状,从而使TensorFlow能够在运行时自动确定Tensor的形状。 总结起来,TensorFlowop是用于执行某种计算或操作的节点,可以接收多个输入Tensor并输出多个输出Tensor。自定义op可以通过注册和定义计算逻辑来扩展TensorFlow的功能。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [tensorflowop](https://blog.csdn.net/wyg_031113/article/details/124517508)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 33.333333333333336%"] - *2* [TensorFlow实现自定义Op方式](https://download.csdn.net/download/weixin_38736760/12855920)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 33.333333333333336%"] - *3* [Deep Learning——TensorFlow中的OP](https://blog.csdn.net/weixin_42067873/article/details/116454518)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 33.333333333333336%"] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值