一篇文章带你搞定 TensorFlow 中的 tf.slice 函数

函数的功能：返回对应的张量的切片，从列表、数组、张量等对象中抽取一部分数据

函数的定义：

tf.slice(inputs, begin, size, name)

inputs：张量
begin：切片的起始位置
size：切片的大小

该函数就是在张量 inputs 中，从 begin 定义的位置开始，获取 size 大小的子集。

示例：

import tensorflow as tf

x = [[1, 2, 3, 4], [5, 6, 7, 8]]
with tf.Session() as sess:
    begin = [1, 0]
    size = [1, 2]
    result = tf.slice(x,begin,size)
    print(sess.run(result))
    print(sess.run(tf.shape(result)))

可以看到，size 表示切片的shape

begin = [1, 0] 表示切片的起始点，比如这里的第一个1，表示 x 的第2行(数组下标从0开始)，这里的0表示从第二行的0下标开始也就是5开始，获取数据。

若 size 中出现 -1 ，表示抽取对应维度上的所有元素：