关于tf.slice()函数

关于tf.slice()函数

网上有很多整个函数的介绍，但看了一遍总觉得各家解释的都不够完善。自己又稍微总结了一下。

tf.slice(tensor,begin,size)

如B=[

   [ [1  1  1]  [2 2 2] ]

   [  [3 3 3]  [4 4 4] ]

   [  [5 5 5]  [6 6 6] ]

                                 ]

上述张量维度为[3,2,3]

如进行A=tf.slice(B,[1,0,0],[1,1,3]) ,可得A=[3 3 3]

计算方式，begin主要确定从何处开始切，size主要确定切多少，理解时，begin确定后住元素后，该元素之后的元素都在待切范围。等size确定切多少后即可确定最终结果。

上述先看begin为[1 0 0 ],第一个1确定第一个范围为[[3 3 3] [4 4 4]] ，第二个0确定为[3 3 3],第三个0确定为[3]，开始位置确定,注意此时整个[[3 3 3] [4 4 4]

        [5 5 5] [6 6 6]

                              ]  都是所需要关注的范围。

再看size[1 1 3 ] ，size中的第一个元素代表取行，第二个代表取列，第三个代表取出之后的维度，第一个1代表取1行，得到[[3 3 3] [4 4 4]]，第二个1代表取1列得[3 3 3]，第三个3代表最终是三个维度得[[[3 3 3]]]

 

同理我们将size换为[1 2 3] 可以得到结果为[[[3 3 3] [4 4 4]]]   

size为[2,1,3]可得结果为[[[3 3 3] [5 5 5]]]

若size内存在-1，则那个维度下的所有数据都会被切走。