tf.slice()是切片分割方法
tf.slice(a,b,c) 其中a是原始数据,b是具体怎么分割,c是得到结果的维度
举个例子
import tensorflow as tf
scale=tf.placeholder(tf.int32)
t1 = [[9, 1, 6], [8, 2, 5],[7, 3, 4], [6, 4, 3],[5, 5, 2], [4, 6, 1]]
d=tf.slice(t1,scale,[3,2])
with tf.Session() as sess:
print(sess.run(d,feed_dict={scale:[3,1]}))
t1是原始数据,是6行3列的
scale是[3,1] 是2个维度,如果t1是三个维度那么scale也是三维的!
[3,2]是最后得到的结果应该是什么维度的;注意最后得到结果的维度与切割数据的维度相加得到的是原始数据的维度
3+3=6 2+1=3 [6,3]列
我们对[6,3]列的数据进行[3,1]切割的意思是,去掉三行,在去掉一列,一般是去掉上面和右方
举个例子
import tensorflow as tf
scale=tf.placeholder(tf.int32)
t1 = [[[9, 1, 6], [8, 2, 5]],[[7, 3, 4], [6, 4, 3]],[[5, 5, 2], [4, 6, 1]]]
d=tf.slice(t1,s