tf.slice函数

tf.slice(
    input_, begin, size, name=None
)

slice函数用于对张量Tensor进行切片，参数begin表示待切片张量的起始索引，参数size表示切割的长度。都用Tensor张量来表示。

1、二维张量的切片

>>> a = tf.reshape(tf.range(9), (3, 3))
>>> a

# [1, 0]表示第一个维度的起始位置为1，第二个维度的起始位置为0
# [2, -1]表示第一个维度的切片大小为2，第二个维度的切片大小为全部
>>> b = tf.slice(a, [1, 0], [2, -1])
>>> b

>>> c = tf.slice(a, [0, 2], [1, 1])
>>> c

2、三维张量的切片

>>> a = tf.constant([[[1, 1, 1], [2, 2, 2]],
		    [[3, 3, 3], [4, 4, 4]],
                    [[5, 5, 5], [6, 6, 6]]])
>>> a

>>> b = tf.slice(a, [0, 0, 0], [1, 2, 3])
>>> b

>>> c = tf.slice(a, [2, 1, 0], [1, 1, 3])
>>> c

>>> d = tf.slice(a, [1, 1, 1], [1, 1, 2])
>>> d