tf.slice(inputs, begin, size, name)
begin和size是两个多维列表决定了要抽取的数据的开始和结束位置
begin表示从inputs的哪个位置元素开始抽取
size表示在inputs的各个维度上抽取的元素数目
size[]中出现-1,表示抽取对应维度上从begin位置开始的所有元素
例1:
with tf.Session() as sess:
x = [[1, 2, 3], [4, 5, 6],[7,8,9],[0,11,12]]
begin = [0,1]
size = [3, 1]
print(sess.run(tf.slice(x,begin,size)))
输出;
[[2]
[5]
[8]]
例2:
with tf.Session() as sess:
x = [[1, 2, 3], [4, 5, 6],[7,8,9],[0,11,12]]
begin = [1,1]
size = [-1, 2]
print(sess.run(tf.slice(x,begin,size)))
输出;
[[ 5 6]
[ 8 9]
[11 12]]