码迷,mamicode.com
首页 > 其他好文 > 详细

TensorFlow基础——常用函数(二)

时间:2019-05-10 13:03:58      阅读:114      评论:0      收藏:0      [点我收藏+]

标签:more   slice   RoCE   pip   单例   must   字符   reads   inpu   

一、tf函数

TensorFlow 将图形定义转换成分布式执行的操作, 以充分利用可用的计算资源(如 CPU 或 GPU。一般你不需要显式指定使用 CPU 还是 GPU, TensorFlow 能自动检测。如果检测到 GPU, TensorFlow 会尽可能地利用找到的第一个 GPU 来执行操作. 并行计算能让代价大的算法计算加速执行,TensorFlow也在实现上对复杂操作进行了有效的改进。大部分核相关的操作都是设备相关的实现,比如GPU。本文主要涉及的相关概念或操作有以下内容:

操作组

操作

Building Graphs

Core graph data structures,Tensor types,Utility functions

Inputs and Readers

Placeholders,Readers,Converting,Queues,Input pipeline

1.1 建立图(Building Graphs)

本节主要介绍建立tensorflow图的相关类或函数

* 核心图的数据结构(Core graph data structures)

tf.Graph

操作

描述

class tf.Graph

tensorflow中的计算以图数据流的方式表示

一个图包含一系列表示计算单元的操作对象

以及在图中流动的数据单元以tensor对象表现

tf.Graph.__init__()

建立一个空图

tf.Graph.as_default()

一个将某图设置为默认图,并返回一个上下文管理器

如果不显式添加一个默认图,系统会自动设置一个全局的默认图。

所设置的默认图,在模块范围内所定义的节点都将默认加入默认图中

tf.Graph.as_graph_def

(from_version=None, add_shapes=False)

返回一个图的序列化的GraphDef表示

序列化的GraphDef可以导入至另一个图中(使用 import_graph_def())

或者使用C++ Session API

tf.Graph.finalize()

完成图的构建,即将其设置为只读模式

tf.Graph.finalized

返回True,如果图被完成

tf.Graph.control_dependencies(control_inputs)

定义一个控制依赖,并返回一个上下文管理器

with g.control_dependencies([a, b, c]):

# `d` 和 `e` 将在 `a`, `b`, 和`c`执行完之后运行.

d = …

e = …

tf.Graph.device(device_name_or_function)

定义运行图所使用的设备,并返回一个上下文管理器

with g.device(‘/gpu:0‘): ...

with g.device(‘/cpu:0‘): ...

tf.Graph.name_scope(name)

为节点创建层次化的名称,并返回一个上下文管理器

tf.Graph.add_to_collection(name, value)

将value以name的名称存储在收集器(collection)中

tf.Graph.get_collection(name, scope=None)

根据name返回一个收集器中所收集的值的列表

tf.Graph.as_graph_element

(obj, allow_tensor=True, allow_operation=True)

返回一个图中与obj相关联的对象,为一个操作节点或者tensor数据

tf.Graph.get_operation_by_name(name)

根据名称返回操作节点

tf.Graph.get_tensor_by_name(name)

根据名称返回tensor数据

tf.Graph.get_operations()

返回图中的操作节点列表

tf.Graph.gradient_override_map(op_type_map)

用于覆盖梯度函数的上下文管理器

 1 #class tf.Graph
 2 #tensorflow运行时需要设置默认的图
 3 g = tf.Graph()
 4 with g.as_default():
 5   # Define operations and tensors in `g`.
 6   c = tf.constant(30.0)
 7   assert c.graph is g
 8 
 9 ##也可以使用tf.get_default_graph()获得默认图,也可在基础上加入节点或子图
10 c = tf.constant(4.0)
11 assert c.graph is tf.get_default_graph()
12 #tf.Graph.as_default
13 #以下两段代码功能相同
14 #1、使用Graph.as_default():
15 g = tf.Graph()
16 with g.as_default():
17   c = tf.constant(5.0)
18   assert c.graph is g
19 
20 #2、构造和设置为默认
21 with tf.Graph().as_default() as g:
22   c = tf.constant(5.0)
23   assert c.graph is g
24 #tf.Graph.control_dependencies(control_inputs)
25 # 错误代码
26 def my_func(pred, tensor):
27   t = tf.matmul(tensor, tensor)
28   with tf.control_dependencies([pred]):
29     # 乘法操作(op)没有创建在该上下文,所以没有被加入依赖控制
30     return t
31 
32 # 正确代码
33 def my_func(pred, tensor):
34   with tf.control_dependencies([pred]):
35     # 乘法操作(op)创建在该上下文,所以被加入依赖控制中
36     #执行完pred之后再执行matmul
37     return tf.matmul(tensor, tensor) 
38 # tf.Graph.name_scope(name)
39 # 一个图中包含有一个名称范围的堆栈,在使用name_scope(...)之后,将压(push)新名称进栈中,
40 #并在下文中使用该名称
41 with tf.Graph().as_default() as g:
42   c = tf.constant(5.0, name="c")
43   assert c.op.name == "c"
44   c_1 = tf.constant(6.0, name="c")
45   assert c_1.op.name == "c_1"
46 
47   # Creates a scope called "nested"
48   with g.name_scope("nested") as scope:
49     nested_c = tf.constant(10.0, name="c")
50     assert nested_c.op.name == "nested/c"
51 
52     # Creates a nested scope called "inner".
53     with g.name_scope("inner"):
54       nested_inner_c = tf.constant(20.0, name="c")
55       assert nested_inner_c.op.name == "nested/inner/c"
56 
57     # Create a nested scope called "inner_1".
58     with g.name_scope("inner"):
59       nested_inner_1_c = tf.constant(30.0, name="c")
60       assert nested_inner_1_c.op.name == "nested/inner_1/c"
61 
62       # Treats `scope` as an absolute name scope, and
63       # switches to the "nested/" scope.
64       with g.name_scope(scope):
65         nested_d = tf.constant(40.0, name="d")
66         assert nested_d.op.name == "nested/d"
67 
68         with g.name_scope(""):
69           e = tf.constant(50.0, name="e")
70           assert e.op.name == "e"
tf.Operation

操作

描述

class tf.Operation

代表图中的一个节点,用于计算tensors数据

该类型将由python节点构造器产生(比如tf.matmul())

或者Graph.create_op()

例如c = tf.matmul(a, b)创建一个Operation类

为类型为”MatMul”,输入为’a’,’b’,输出为’c’的操作类

tf.Operation.name

操作节点(op)的名称

tf.Operation.type

操作节点(op)的类型,比如”MatMul”

tf.Operation.inputs

tf.Operation.outputs

操作节点的输入与输出

tf.Operation.control_inputs

操作节点的依赖

tf.Operation.run(feed_dict=None, session=None)

在会话(Session)中运行该操作

tf.Operation.get_attr(name)

获取op的属性值

tf.Tensor

操作

描述

class tf.Tensor

表示一个由操作节点op产生的值,

TensorFlow程序使用tensor数据结构来代表所有的数据, 

计算图中, 操作间传递的数据都是 tensor,一个tensor是一个符号handle,

里面并没有表示实际数据,而相当于数据流的载体

tf.Tensor.dtype

tensor中数据类型

tf.Tensor.name

该tensor名称

tf.Tensor.value_index

该tensor输出外op的index

tf.Tensor.graph

该tensor所处在的图

tf.Tensor.op

产生该tensor的op

tf.Tensor.consumers()

返回使用该tensor的op列表

tf.Tensor.eval(feed_dict=None, session=None)

在会话中求tensor的值

需要使用with sess.as_default()或者 eval(session=sess)

tf.Tensor.get_shape()

返回用于表示tensor的shape的类TensorShape

tf.Tensor.set_shape(shape)

更新tensor的shape

tf.Tensor.device

设置计算该tensor的设备

 1 #tf.Tensor.get_shape()
 2 c = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
 3 print(c.get_shape())
 4 ==> TensorShape([Dimension(2), Dimension(3)])
 5  
 6 #现在有个用于图像处理的tensor->image
 7 print(image.get_shape())
 8 ==> TensorShape([Dimension(None), Dimension(None), Dimension(3)])
 9 # 假如我们知道数据集中图像尺寸为28 x 28,那么可以设置
10 image.set_shape([28, 28, 3])
11 print(image.get_shape())
12 ==> TensorShape([Dimension(28), Dimension(28), Dimension(3)])

 

* tensor类型(Tensor types)

tf.DType

操作

描述

class tf.DType

数据类型主要包含

tf.float16,tf.float16,tf.float32,tf.float64,

tf.bfloat16,tf.complex64,tf.complex128,

tf.int8,tf.uint8,tf.uint16,tf.int16,tf.int32,

tf.int64,tf.bool,tf.string

tf.DType.is_compatible_with(other)

判断other的数据类型是否将转变为该DType

tf.DType.name

数据类型名称

tf.DType.base_dtype

返回该DType的基础DType,而非参考的数据类型(non-reference)

tf.DType.as_ref

返回一个基于DType的参考数据类型

tf.DType.is_floating

判断是否为浮点类型

tf.DType.is_complex

判断是否为复数

tf.DType.is_integer

判断是否为整数

tf.DType.is_unsigned

判断是否为无符号型数据

tf.DType.as_numpy_dtype

返回一个基于DType的numpy.dtype类型

tf.DType.max

tf.DType.min

返回这种数据类型能表示的最大值及其最小值

tf.as_dtype(type_value)

返回由type_value转变得的相应tf数据类型

 

通用函数(Utility functions)

操作

描述

tf.device(device_name_or_function)

基于默认的图,其功能便为Graph.device()

tf.container(container_name)

基于默认的图,其功能便为Graph.container()

tf.name_scope(name)

基于默认的图,其功能便为 Graph.name_scope()

tf.control_dependencies(control_inputs)

基于默认的图,其功能便为Graph.control_dependencies()

tf.convert_to_tensor
(value, dtype=None, name=None, as_ref=False)

将value转变为tensor数据类型

tf.get_default_graph()

返回返回当前线程的默认图

tf.reset_default_graph()

清除默认图的堆栈,并设置全局图为默认图

tf.import_graph_def(graph_def, input_map=None,

return_elements=None, name=None, op_dict=None,

producer_op_list=None)

将graph_def的图导入到python中

 

图收集(Graph collections)

操作

描述

tf.add_to_collection(name, value)

基于默认的图,其功能便为Graph.add_to_collection()

tf.get_collection(key, scope=None)

基于默认的图,其功能便为Graph.get_collection()

定义新操作节点(Defining new operations)

tf.RegisterGradient

操作

描述

class tf.RegisterGradient

返回一个用于寄存op类型的梯度函数的装饰器

tf.NoGradient(op_type)

设置操作节点类型op_type的节点没有指定的梯度

class tf.RegisterShape

返回一个用于寄存op类型的shape函数的装饰器

class tf.TensorShape

表示tensor的shape

tf.TensorShape.merge_with(other)

与other合并shape信息,返回一个TensorShape类

tf.TensorShape.concatenate(other)

与other的维度相连结

tf.TensorShape.ndims

返回tensor的rank

tf.TensorShape.dims

返回tensor的维度

tf.TensorShape.as_list()

以list的形式返回tensor的shape

tf.TensorShape.is_compatible_with(other)

判断shape是否为兼容

TensorShape(None)与其他任何shape值兼容

class tf.Dimension

 

tf.Dimension.is_compatible_with(other)

判断dims是否为兼容

tf.Dimension.merge_with(other)

与other合并dims信息

tf.op_scope(values, name, default_name=None)

在python定义op时,返回一个上下文管理器

 1 #tf.RegisterGradient
 2 #该装饰器只使用于定义一个新的op类型时候,如果一个op有m个输入,n个输出。那么该梯度函数应该设置原始的
 3 #操作类型,以及n个Tensor对象(表示每一个op输出的梯度),以及m个对象(表示每一个op输入的偏梯度)
 4 #以操作节点类型为‘Sub‘为例,两输入为x,y。为一个输出x-y
 5 @tf.RegisterGradient("Sub")
 6 def _sub_grad(unused_op, grad):
 7   return grad, tf.neg(grad)
 8  
 9 
10 #tf.op_scope
11 #定义一个名称为my_op的python操作节点op
12 def my_op(a, b, c, name=None):
13   with tf.op_scope([a, b, c], name, "MyOp") as scope:
14     a = tf.convert_to_tensor(a, name="a")
15     b = tf.convert_to_tensor(b, name="b")
16     c = tf.convert_to_tensor(c, name="c")
17     # Define some computation that uses `a`, `b`, and `c`.
18     return foo_op(..., name=scope)

 

 1.2 输入和读取器(Inputs and Readers)

本节主要介绍tensorflow中数据的读入相关类或函数

* 占位符(Placeholders)

tf提供一种占位符操作,在执行时需要为其提供数据data。

操作

描述

tf.placeholder(dtype, shape=None, name=None)

为一个tensor插入一个占位符

eg:x = tf.placeholder(tf.float32, shape=(1024, 1024))

tf.placeholder_with_default(input, shape, name=None)

当输出没有fed时,input通过一个占位符op

tf.sparse_placeholder(dtype, shape=None, name=None)

为一个稀疏tensor插入一个占位符


* 读取器(Readers)

tf提供一系列读取各种数据格式的类。对于多文件输入,可以使用函数tf.train.string_input_producer,该函数将创建一个保持文件的FIFO队列,以供reader使用。或者如果输入的这些文件名有相雷同的字符串,也可以使用函数tf.train.match_filenames_once。

操作

描述

class tf.ReaderBase

不同的读取器类型的基本类

tf.ReaderBase.read(queue, name=None)

返回下一个记录对(key, value),queue为tf文件队列FIFOQueue

tf.ReaderBase.read_up_to(queue, num_records, name=None)

返回reader产生的num_records对(key, value)

tf.ReaderBase.reader_ref

返回应用在该reader上的Op

tf.ReaderBase.reset(name=None)

恢复reader为初始状态

tf.ReaderBase.restore_state(state, name=None)

恢复reader为之前的保存状态state

tf.ReaderBase.serialize_state(name=None)

返回一个reader解码后产生的字符串tansor

class tf.TextLineReader

 

tf.TextLineReader.num_records_produced(name=None)

返回reader已经产生的记录(records )数目

tf.TextLineReader.num_work_units_completed(name=None)

返回该reader已经完成的处理的work数目

tf.TextLineReader.read(queue, name=None)

返回reader所产生的下一个记录对 (key, value),该reader可以限定新产生输出的行数

tf.TextLineReader.reader_ref

返回应用在该reader上的Op

tf.TextLineReader.reset(name=None)

恢复reader为初始状态

tf.TextLineReader.restore_state(state, name=None)

恢复reader为之前的保存状态state

tf.TextLineReader.serialize_state(name=None)

返回一个reader解码后产生的字符串tansor

class tf.WholeFileReader

一个阅读器,读取整个文件,返回文件名称key,以及文件中所有的内容value,该类的方法同上,不赘述

class tf.IdentityReader

一个reader,以key和value的形式,输出一个work队列。该类其他方法基本同上

class tf.TFRecordReader

读取TFRecord格式文件的reader。该类其他方法基本同上

class tf.FixedLengthRecordReader

输出


* 数据转换(Converting)

tf提供一系列方法将各种格式数据转换为tensor表示。

操作

描述

tf.decode_csv(records, record_defaults, 

field_delim=None, name=None)

将csv转换为tensor,与tf.TextLineReader搭配使用

tf.decode_raw(bytes, out_type, 

little_endian=None, name=None)

将bytes转换为一个数字向量表示,bytes为一个字符串类型的tensor

与函数 tf.FixedLengthRecordReader搭配使用,详见tf的CIFAR-10例子

选取与要输入的文件格式相匹配的reader,并将文件队列提供给reader的读方法( read method)。读方法将返回文件唯一标识的key,以及一个记录(record)(有助于对出现一些另类的records时debug),以及一个标量的字符串值。再使用一个(或多个)解码器(decoder) 或转换操作(conversion ops)将字符串转换为tensor类型。

 1 #读取文件队列,使用reader中read的方法,返回key与value
 2 filename_queue = tf.train.string_input_producer(["file0.csv", "file1.csv"])
 3 reader = tf.TextLineReader()
 4 key, value = reader.read(filename_queue)
 5 
 6 record_defaults = [[1], [1], [1], [1], [1]]
 7 col1, col2, col3, col4, col5 = tf.decode_csv(
 8     value, record_defaults=record_defaults)
 9 features = tf.pack([col1, col2, col3, col4])
10 
11 with tf.Session() as sess:
12   # Start populating the filename queue.
13   coord = tf.train.Coordinator()
14   threads = tf.train.start_queue_runners(coord=coord)
15 
16   for i in range(1200):
17     # Retrieve a single instance:
18     example, label = sess.run([features, col5])
19 
20   coord.request_stop()
21   coord.join(threads)

* Example protocol buffer

提供了一些Example protocol buffers,tf所推荐的用于训练样本的数据格式,它们包含特征信息,详情可见。 
这是一种与前述将手上现有的各种数据类型转换为支持的格式的方法,这种方法更容易将网络结构与数据集融合或匹配。这种tensorflow所推荐的数据格式是一个包含tf.train.Example protocol buffers (包含特征Features域)的TFRecords文件。 
1、获取这种格式的文件方式为,首先将一般的数据格式填入Example protocol buffer中,再将 protocol buffer序列化为一个字符串,然后使用tf.python_io.TFRecordWriter类的相关方法将字符串写入一个TFRecords文件中,参见MNIST例子,将MNIST 数据转换为该类型数据。 
2、读取TFRecords格式文件的方法为,使用tf.TFRecordReader读取器和tf.parse_single_example解码器。parse_single_example操作将 example protocol buffers解码为tensor形式。参见MNIST例子

操作

描述

class tf.VarLenFeature

解析变长的输入特征feature相关配置

class tf.FixedLenFeature

解析定长的输入特征feature相关配置

class tf.FixedLenSequenceFeature

序列项目中的稠密(dense )输入特征的相关配置

tf.parse_example(serialized, features, 
name=None, example_names=None)

将一组Example protos解析为tensor的字典形式

解析serialized所给予的序列化的一些Example protos

返回一个由特征keys映射Tensor和SparseTensor值的字典

tf.parse_single_example(serialized, features, 
name=None, example_names=None)

解析一个单独的Example proto,与tf.parse_example方法雷同

tf.decode_json_example(json_examples, name=None)

将JSON编码的样本记录转换为二进制protocol buffer字符串

  1 #tf.parse_example的使用举例
  2 #输入序列化数据如下: 
  3 serialized = [
  4   features
  5     { feature { key: "ft" value { float_list { value: [1.0, 2.0] } } } },
  6   features
  7     { feature []},
  8   features
  9     { feature { key: "ft" value { float_list { value: [3.0] } } }
 10 ]
 11 #那么输出为一个字典(dict),如下:
 12 {"ft": SparseTensor(indices=[[0, 0], [0, 1], [2, 0]],
 13                     values=[1.0, 2.0, 3.0],
 14                     shape=(3, 2)) }
 15 #########
 16 #再来看一个例子,给定两个序列化的原始输入样本:
 17 [
 18   features {
 19     feature { key: "kw" value { bytes_list { value: [ "knit", "big" ] } } }
 20     feature { key: "gps" value { float_list { value: [] } } }
 21   },
 22   features {
 23     feature { key: "kw" value { bytes_list { value: [ "emmy" ] } } }
 24     feature { key: "dank" value { int64_list { value: [ 42 ] } } }
 25     feature { key: "gps" value { } }
 26   }
 27 ]
 28 #相关参数如下:
 29 example_names: ["input0", "input1"],
 30 features: {
 31     "kw": VarLenFeature(tf.string),
 32     "dank": VarLenFeature(tf.int64),
 33     "gps": VarLenFeature(tf.float32),
 34 }
 35 #那么有如下输出:
 36 {
 37   "kw": SparseTensor(
 38       indices=[[0, 0], [0, 1], [1, 0]],
 39       values=["knit", "big", "emmy"]
 40       shape=[2, 2]),
 41   "dank": SparseTensor(
 42       indices=[[1, 0]],
 43       values=[42],
 44       shape=[2, 1]),
 45   "gps": SparseTensor(
 46       indices=[],
 47       values=[],
 48       shape=[2, 0]),
 49 }
 50 #########
 51 #对于两个样本的输出稠密结果情况
 52 [
 53   features {
 54     feature { key: "age" value { int64_list { value: [ 0 ] } } }
 55     feature { key: "gender" value { bytes_list { value: [ "f" ] } } }
 56    },
 57    features {
 58     feature { key: "age" value { int64_list { value: [] } } }
 59     feature { key: "gender" value { bytes_list { value: [ "f" ] } } }
 60   }
 61 ]
 62 #我们可以使用以下参数
 63 example_names: ["input0", "input1"],
 64 features: {
 65     "age": FixedLenFeature([], dtype=tf.int64, default_value=-1),
 66     "gender": FixedLenFeature([], dtype=tf.string),
 67 }
 68 #期望的结果如下
 69 {
 70   "age": [[0], [-1]],
 71   "gender": [["f"], ["f"]],
 72 }
 73  
 74 
 75 ##Example protocol buffer相关使用的例子
 76 #将mnist的数据转换为TFRecords文件格式
 77 import os
 78 import tensorflow as tf
 79 from tensorflow.contrib.learn.python.learn.datasets import mnist
 80 SOURCE_URL = http://yann.lecun.com/exdb/mnist/
 81 
 82 TRAIN_IMAGES = train-images-idx3-ubyte.gz  # MNIST filenames
 83 TRAIN_LABELS = train-labels-idx1-ubyte.gz
 84 TEST_IMAGES = t10k-images-idx3-ubyte.gz
 85 TEST_LABELS = t10k-labels-idx1-ubyte.gz
 86 
 87 tf.app.flags.DEFINE_string(directory, /tmp/data,
 88                            Directory to download data files and write the 
 89                            converted result)
 90 tf.app.flags.DEFINE_integer(validation_size, 5000,
 91                             Number of examples to separate from the training 
 92                             data for the validation set.)
 93 FLAGS = tf.app.flags.FLAGS
 94 
 95 def _int64_feature(value):
 96   return tf.train.Feature(int64_list=tf.train.Int64List(value=[value]))
 97 
 98 def _bytes_feature(value):
 99   return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]))
100 
101 def convert_to(data_set, name):
102   images = data_set.images
103   labels = data_set.labels
104   num_examples = data_set.num_examples
105 
106   if images.shape[0] != num_examples:
107     raise ValueError(Images size %d does not match label size %d. %
108                      (images.shape[0], num_examples))
109   rows = images.shape[1]
110   cols = images.shape[2]
111   depth = images.shape[3]
112 
113   filename = os.path.join(FLAGS.directory, name + .tfrecords)
114   print(Writing, filename)
115   writer = tf.python_io.TFRecordWriter(filename)
116   for index in range(num_examples):
117     image_raw = images[index].tostring()
118     example = tf.train.Example(features=tf.train.Features(feature={
119         height: _int64_feature(rows),
120         width: _int64_feature(cols),
121         depth: _int64_feature(depth),
122         label: _int64_feature(int(labels[index])),
123         image_raw: _bytes_feature(image_raw)}))
124     writer.write(example.SerializeToString())
125   writer.close()
126 
127 def main(argv):
128   # Get the data.
129   data_sets = mnist.read_data_sets(FLAGS.directory,
130                                    dtype=tf.uint8,
131                                    reshape=False)
132 
133   # Convert to Examples and write the result to TFRecords.
134   convert_to(data_sets.train, train)
135   convert_to(data_sets.validation, validation)
136   convert_to(data_sets.test, test)
137 
138 if __name__ == __main__:
139   tf.app.run()

 

* 队列(Queues)

tensorflow提供了几个队列应用,用来将tf计算图与tensors的阶段流水组织到一起。队列是使用tensorflow计算的一个强大的机制,正如其他Tensorflow的元素一样,一个队列也是tf图中的一个节点(node),它是一个有状态的node,就像一个变量:其他节点可以改变其内容。 
我们来看一个简单的例子,如下gif图,我们将创建一个先入先出队列(FIFOQueue)并且将值全设为0,然后我们构建一个图以获取队列出来的元素,对该元素加1操作,并将结果再放入队列末尾。渐渐地,队列中的数字便增加。 
技术图片

操作

描述

class tf.QueueBase

基本的队列应用类.队列(queue)是一种数据结构,

该结构通过多个步骤存储tensors,

并且对tensors进行入列(enqueue)与出列(dequeue)操作

tf.QueueBase.enqueue(vals, name=None)

将一个元素编入该队列中。如果在执行该操作时队列已满,

那么将会阻塞直到元素编入队列之中

tf.QueueBase.enqueue_many(vals, name=None)

将零个或多个元素编入该队列中

tf.QueueBase.dequeue(name=None)

将元素从队列中移出。如果在执行该操作时队列已空,

那么将会阻塞直到元素出列,返回出列的tensors的tuple

tf.QueueBase.dequeue_many(n, name=None)

将一个或多个元素从队列中移出

tf.QueueBase.size(name=None)

计算队列中的元素个数

tf.QueueBase.close

(cancel_pending_enqueues=False, name=None)

关闭该队列

f.QueueBase.dequeue_up_to(n, name=None)

从该队列中移出n个元素并将之连接

tf.QueueBase.dtypes

列出组成元素的数据类型

tf.QueueBase.from_list(index, queues)

根据queues[index]的参考队列创建一个队列

tf.QueueBase.name

返回最队列下面元素的名称

tf.QueueBase.names

返回队列每一个组成部分的名称

class tf.FIFOQueue

在出列时依照先入先出顺序,其他方法与tf.QueueBase雷同

class tf.PaddingFIFOQueue

一个FIFOQueue ,同时根据padding支持batching变长的tensor

class tf.RandomShuffleQueue

该队列将随机元素出列,其他方法与tf.QueueBase雷同


* 文件系统的处理(Dealing with the filesystem)

操作

描述

tf.matching_files(pattern, name=None)

返回与pattern匹配模式的文件名称

tf.read_file(filename, name=None)

读取并输出输入文件的整个内容


* 输入管道(Input pipeline)

用于设置输入预取数的管道TF函数,函数 “producer”添加一个队列至图中,同时一个相应用于运行队列中子图(subgraph)的QueueRunner

操作

描述

tf.train.match_filenames_once(pattern, name=None)

保存与pattern的文件列表

tf.train.limit_epochs(tensor, num_epochs=None, name=None)

返回一个num_epochs次数,然后报告OutOfRange错误

tf.train.input_producer(input_tensor, element_shape=None, 

num_epochs=None, shuffle=True, seed=None, capacity=32, 

shared_name=None, summary_name=None, name=None)

为一个输入管道输出input_tensor中的多行至一个队列中

tf.train.range_input_producer(limit, num_epochs=None, 

shuffle=True, seed=None, capacity=32, 

shared_name=None, name=None)

产生一个从1至limit-1的整数至队列中

tf.train.slice_input_producer(tensor_list, num_epochs=None, 

shuffle=True, seed=None, capacity=32, 

shared_name=None, name=None)

对tensor_list中的每一个tensor切片

tf.train.string_input_producer(string_tensor, num_epochs=None,
shuffle=True, seed=None, capacity=32, 
shared_name=None, name=None)

为一个输入管道输出一组字符串(比如文件名)至队列中


* 在输入管道末端批量打包(Batching at the end of an input pipeline)

该相关函数增添一个队列至图中以将数据一样本打包为batch。它们也会添加 一个QueueRunner,以便执行的已经被填满队列的子图

操作

描述

tf.train.batch(tensors, batch_size, num_threads=1,

capacity=32, enqueue_many=False, shapes=None, 

dynamic_pad=False, allow_smaller_final_batch=False, 

shared_name=None, name=None)

在输入的tensors中创建一些tensor数据格式的batch,

若输入为shape[*, x, y, z],那么输出则为[batch_size, x, y, z]

返回一个列表或者一个具有与输入tensors相同类型tensors的字典

tf.train.batch_join(tensors_list, batch_size, 

capacity=32, enqueue_many=False, shapes=None, 

dynamic_pad=False, allow_smaller_final_batch=False, 

shared_name=None, name=None)

将一个tensors的列表添加至一个队列中以创建样本的batches

len(tensors_list)个线程将启动,

线程i将tensors_list[i]的tensors入列

tensors_list[i1][j]与tensors_list[i2][j]有相同的类型和shape

tf.train.shuffle_batch(tensors, batch_size, capacity, 
min_after_dequeue, num_threads=1, seed=None, 
enqueue_many=False, shapes=None, 
allow_smaller_final_batch=False,
shared_name=None, name=None)

使用随机乱序的方法创建batches

tensors:用于入列的一个list或者dict

capacity:一个整数,表示队列中元素最大数目

tf.train.shuffle_batch_join(tensors_list, batch_size, 

capacity, min_after_dequeue, seed=None, 

enqueue_many=False, shapes=None, 

allow_smaller_final_batch=False, 

shared_name=None, name=None)

随机乱序的tensors创建batches,

其中tensors_list参数为tensors元组或tensors字典的列表

len(tensors_list)个线程将启动,

线程i将tensors_list[i]的tensors入列

tensors_list[i1][j]与tensors_list[i2][j]有相同的类型和shape

 1 # 一个简单例子,使用tf.train.shuffle_batch创建一个具有32张图像和32个标签的batches.
 2 image_batch, label_batch = tf.train.shuffle_batch(
 3       [single_image, single_label],
 4       batch_size=32,
 5       num_threads=4,
 6       capacity=50000,
 7       min_after_dequeue=10000)
 8  
 9 #Batching函数相关例子,以函数tf.train.shuffle_batch为例
10 #为training, evaluation等操作将样本batching,以下代码使用随机顺序打包样本
11 def read_my_file_format(filename_queue):
12   reader = tf.SomeReader()
13   key, record_string = reader.read(filename_queue)
14   example, label = tf.some_decoder(record_string)
15   processed_example = some_processing(example)
16   return processed_example, label
17 
18 def input_pipeline(filenames, batch_size, num_epochs=None):
19   filename_queue = tf.train.string_input_producer(
20       filenames, num_epochs=num_epochs, shuffle=True)
21   example, label = read_my_file_format(filename_queue)
22   # min_after_dequeue defines how big a buffer we will randomly sample
23   #   from -- bigger means better shuffling but slower start up and more
24   #   memory used.
25   # capacity must be larger than min_after_dequeue and the amount larger
26   #   determines the maximum we will prefetch.  Recommendation:
27   #   min_after_dequeue + (num_threads + a small safety margin) * batch_size
28   min_after_dequeue = 10000
29   capacity = min_after_dequeue + 3 * batch_size
30   example_batch, label_batch = tf.train.shuffle_batch(
31       [example, label], batch_size=batch_size, capacity=capacity,
32       min_after_dequeue=min_after_dequeue)
33   return example_batch, label_batch
34  
35 
36 #如果需要跟多的并行或文件之间的样本乱序操作,可以使用函数tf.train.shuffle_batch_join多实例化reader
37 def read_my_file_format(filename_queue):
38   # 与上例子相同
39 
40 def input_pipeline(filenames, batch_size, read_threads, num_epochs=None):
41   filename_queue = tf.train.string_input_producer(
42       filenames, num_epochs=num_epochs, shuffle=True)
43   example_list = [read_my_file_format(filename_queue)
44                   for _ in range(read_threads)]
45   min_after_dequeue = 10000
46   capacity = min_after_dequeue + 3 * batch_size
47   example_batch, label_batch = tf.train.shuffle_batch_join(
48       example_list, batch_size=batch_size, capacity=capacity,
49       min_after_dequeue=min_after_dequeue)

 

TensorFlow基础——常用函数(二)

标签:more   slice   RoCE   pip   单例   must   字符   reads   inpu   

原文地址:https://www.cnblogs.com/zhangchao162/p/10843555.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!