Tensorflow--矩阵切片与连接

博客转载自：https://blog.csdn.net/davincil/article/details/77893185

 

函数原型：slice(input_, begin, size, name=None) 
参数： 
input：待切片的矩阵tensor。 
begin：起始位置，表示从哪一个数据开始进行切片。这个起始位置从0开始。若input是一个n维的矩阵，则begin是一个长度为n的tensor。 
size：切片的大小（尺寸），表示则起始位置开始获取每一维上的若干数据。是一个长度与begin相同的tensor。若size中第n个的数据为-1，则表示在该维度上，从起始位置开始的所有数据均被返回。 
name：该操作的名称，是一个可选参数，默认为None。

对于一个n维的矩阵，需满足如下关系： 
0 <= begin[i] <= begin[i] + size[i] <= Di for i in [0, n]

import tensorflow as tf # Tensorflow交互式会话
tf.InteractiveSession() # 定义5x5大小的一个矩阵变量
a = tf.Variable(tf.truncated_normal(shape=[5, 5], dtype=tf.float32)) # 进行切片操作，起始位置为[1,1]（从0开始），大小[2,2]
b = tf.slice(a, [1, 1], [2, 2]) # 同上
c = tf.Variable(tf.truncated_normal(shape=[2, 6, 5], dtype=tf.float32)) d = tf.slice(c, [0, 2, 3], [2, 3, 1]) # 全局变量初始化
tf.global_variables_initializer().run() # 输出
print("Example 01") print("the original matrix:\n", a.eval()) print("after being sliced:\n", b.eval()) print("Example 02") print("the original matrix:\n", c.eval()) print("after being sliced:\n", d.eval())

 

程序运行结果如下：（结果或有不同）

Example 01 the original matrix: [[ 1.37798977  0.27846026  0.07193759  0.44368556  0.65868556] [-0.57639289 -0.64335102 -0.62483543  0.38987917  0.29301718] [ 0.18187736  0.11397317  1.85999572 -0.26037475  0.98114467] [ 0.69557261  0.01183218 -0.27376401 -1.15162456  1.11336803] [-0.66582751 -0.04991583 -1.58189285  0.98189503 -1.11317801]] after being sliced: [[-0.64335102 -0.62483543] [ 0.11397317  1.85999572]] Example 02 the original matrix: [[[-0.44467756 -1.05340731 -0.32313645 -0.69316941  0.04659459] [ 0.01275753 -0.11907347  1.70015264  0.60470396 -0.23756829] [ 0.07424127  1.01376414 -1.15661514 -0.46597373 -1.82189155] [-0.66635352 -0.34318891  0.49555108  0.13062055 -0.67137426] [ 0.04240284  0.55397838 -0.09988129 -0.93551743  0.6810317 ] [ 1.06745911  0.49900523  1.0482769   0.39871195  1.23199737]] [[ 1.22305858 -0.839634    0.63722724 -1.39846325 -0.04114933] [-1.11448932  0.20783874  0.39737079  1.13769484 -0.09408376] [-0.66636425  0.37878662 -0.32013494 -0.26526076  1.53422773] [-0.55344075  0.23021726  0.10251451  0.08433547  1.19850338] [ 1.73070538 -0.50309545 -0.52816319 -0.41802529 -1.52679396] [-1.60076332  0.88759929  0.01327948 -0.7242741  -0.70737672]]] after being sliced: [[[-0.46597373] [ 0.13062055] [-0.93551743]] [[-0.26526076] [ 0.08433547] [-0.41802529]]]

 

2、TensorFlow矩阵链接操作：tf.concat函数 
函数原型：concat(values, axis, name=”concat”) 
参数： 
values：需要链接的矩阵的集合，通常可以是一个list。 
axis：需要进行链接的维度，若矩阵是n维的，则axis的取值为0~n-1。 
name：名称，是一个可选参数。

import tensorflow as tf # Tensorflow交互式会话
tf.InteractiveSession() # 定义两个矩阵，大小为2x3x4
a = tf.Variable(tf.truncated_normal(shape=[2,3,4], dtype=tf.float32)) b = tf.Variable(tf.truncated_normal(shape=[2,3,4], dtype=tf.float32)) # 按照维度0链接
c1 = tf.concat([a, b], axis=0) # 按照维度1链接
c2 = tf.concat([a, b], axis=1) # 按照维度2链接
c3 = tf.concat([a, b], axis=2) # 初始化变量
tf.global_variables_initializer().run() # 输出
print("01") print(c1) print(c1.eval()) print("02") print(c2) print(c2.eval()) print("03") print(c3) print(c3.eval())

 

程序运行结果如下：

Tensor("concat:0", shape=(4, 3, 4), dtype=float32) [[[-0.08826777  1.92810595 -0.79408133 -0.34322619] [-1.71443737  0.70375884 -0.78194672 -0.41254947] [ 0.89348751 -0.08941202  0.70108914  0.64701825]] [[ 1.50688016  0.45680258 -1.08100998  0.24127837] [ 0.58221173 -1.41846514 -1.63450527 -0.41922286] [ 0.48436531 -1.20013559  0.95647675 -0.03131635]] [[-0.03254275 -1.8339541  -0.81978613 -1.25303519] [-1.55067682 -0.37825376 -0.63578284 -0.83120823] [ 0.09672505 -0.43550658 -0.31754431 -0.37109831]] [[ 1.59722102 -0.32856748 -1.33017409  1.43195128] [-0.58259052 -1.60538054  0.07504115  0.8916716 ] [-1.23682356 -0.24931362  1.19812703 -0.81907171]]] 02 Tensor("concat_1:0", shape=(2, 6, 4), dtype=float32) [[[-0.08826777  1.92810595 -0.79408133 -0.34322619] [-1.71443737  0.70375884 -0.78194672 -0.41254947] [ 0.89348751 -0.08941202  0.70108914  0.64701825] [-0.03254275 -1.8339541  -0.81978613 -1.25303519] [-1.55067682 -0.37825376 -0.63578284 -0.83120823] [ 0.09672505 -0.43550658 -0.31754431 -0.37109831]] [[ 1.50688016  0.45680258 -1.08100998  0.24127837] [ 0.58221173 -1.41846514 -1.63450527 -0.41922286] [ 0.48436531 -1.20013559  0.95647675 -0.03131635] [ 1.59722102 -0.32856748 -1.33017409  1.43195128] [-0.58259052 -1.60538054  0.07504115  0.8916716 ] [-1.23682356 -0.24931362  1.19812703 -0.81907171]]] 03 Tensor("concat_2:0", shape=(2, 3, 8), dtype=float32) [[[-0.08826777  1.92810595 -0.79408133 -0.34322619 -0.03254275 -1.8339541
   -0.81978613 -1.25303519] [-1.71443737  0.70375884 -0.78194672 -0.41254947 -1.55067682 -0.37825376
   -0.63578284 -0.83120823] [ 0.89348751 -0.08941202  0.70108914  0.64701825  0.09672505 -0.43550658
   -0.31754431 -0.37109831]] [[ 1.50688016  0.45680258 -1.08100998  0.24127837  1.59722102 -0.32856748
   -1.33017409  1.43195128] [ 0.58221173 -1.41846514 -1.63450527 -0.41922286 -0.58259052 -1.60538054
    0.07504115  0.8916716 ] [ 0.48436531 -1.20013559  0.95647675 -0.03131635 -1.23682356 -0.24931362
    1.19812703 -0.81907171]]]