关于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,则那个维度下的所有数据都会被切走。