TensorFlow基础知识3-操作

xiaoxiao2021-02-28  26

1运算操作的相关知识点

一个运算操作代表了一种类型的抽象运算,比如矩阵乘法或者向量加法。 一个运算操作可以有自己的属性,但是所有属性都必须被预先设置,或者能够在创建计算图时根据上下文推断出来。 通过设置运算操作的属性可以用来支持不同的tensor元素类型,比如让向量加法支持浮点或者整数。 运算核(kernel)是一个运算操作在某个具体的硬件(比如cpu或者gpu中)的实现。 在TensorFlow中,可以通过注册机制加入新的运算操作或者为已有的运算操作添加新的计算核。

与操作相关的API均位于tf.Operation类中

操作描述tf.Operation.name操作的名称tf.Operation.type操作的类型,如MatMultf.Operation.inputs操作的输入tf.Operation.outputs操作的输出tf.Operation.run在会话中运行该操作tf.Operation.get_attr获取操作的属性值

TensorFlow内建运算操作

运算类型运算示例标量运算add、sub、mul、div、exp、log、greater、less、equal向量运算concat、slice、split、constant、rank、shape、shuffle矩阵运算matmul、matricinverse、matrixdeterminant带状态的运算variable、assign、assignadd神经网络组件softmax、sigmoid、relu、convolution2D、maxpooling储存,恢复save、restore队列及同步运算enqueue、dequeue、mutexAcquire、mutexrelease控制流merge、switch、enter、leave、nextInteration

2操作的使用

下图显示了一个数据流图:

根据数据流图写出代码如下:

import tensorflow as tf a = tf.constant(5, name="input_a") b = tf.constant(3, name="input_b") c = tf.multiply(a, b, name="mul_c") d = tf.add(a, b, name="add_d") e = tf.add(c, d, name="add_e") 123456

通过会话运行最后结果e

sess = tf.Session() sess.run(e) 12

输出:23


a = tf.constant(5, name="input_a") b = tf.constant(3, name="input_b") 12

节点a和b,引用了Operation:tf.constant()。在TensorFlow中,数据流图中的每个节点都被称为一个Operation(简记Op)

c = tf.multiply(a, b, name="mul_c") 1

节点c使用了tf.multiply Op,节点c接收两个输入a和b,然后将它们的乘积输出。

3运算符重载

运算符重载 TensorFlow还对常见数学运算符进行了重载,以使乘法、加法、减法及其他常见运算更加简洁。如果运算符有一个或多个参数(操作对象)为Tensor对象,则会有一个TensorFlow Op被调用,并被添加到数据流图中。例如,可按照下列方式轻松地实现两个张量的加法: 假设a和b均为Tensor对象,且形状匹配

下面给出可用于张量的重载运算符的完整清单。

一元运算符

运算符相关TensorFlow运算描述-xtf.neg()返回x中每个元素的相反数~xtf.logical_not()返回x中每个元素的逻辑非。只适用于dtype为tf.bool的Tensor对象abs(x)tf.abs()返回x中每个元素的绝对值

二元运算符

运算符相关TensorFlow运算描述x+ytf.add()将x和y逐元素相加x-ytf.sub()将x和y逐元素相减x*ytf.mul()将x和y逐元素相乘x/y(Python2.x)tf.div()给定整数张量时,执行逐元素的整数除法;给定浮点型张量时,将执行浮点数(“真正的”)除法x/y(Python3.x)tf.truediv()逐元素的浮点数除法(包括分子分母为整数的情形)x//y(Python3.x)tf.floordiv()逐元素的向下取整除法,不返回余数x%ytf.mod()逐元素取模x**ytf.pow()逐一计算x中的每个元素为底数,y中相应元素为指数时的幂xtf.less()逐元素地计算x<y的真值表x<=ytf.less_equal()逐元素地计算x≤y的真值表x>ytf.greater()逐元素地计算x>y的真值表x>=ytf.greater_equal()逐元素地计算x≥y的真值表x&ytf.logical_and()逐元素地计算x & y的真值表,每个元素的dtype属性必须为tf.boolx|ytf.logical_or()逐元素地计算x|y的真值表,每个元素的dtype属性必须为tf.boolx^ytf.logical_xor()逐元素地计算x^y的真值表,每个元素的dtype属性必须为tf.bool

利用这些重载运算符可快速地对代码进行整合,但却无法为这些Op指定name值。如果需要为Op指定name值,请直接调用TensorFlow Op。 

从技术角度讲,==运算符也被重载了,但它不会返回一个布尔型的Tensor对象。它所判断的是两个Tensor对象名是否引用了同一个对象,若是,则返回True,否则,返回False。这个功能主要是在TensorFlow内部使用。如果希望检查张量值是否相同,请使用tf.equal()和tf.not_equal()。

文章出处:https://blog.csdn.net/hongxue8888/article/details/76597322

转载请注明原文地址: https://www.6miu.com/read-2624220.html

最新回复(0)