功能
如果把Tensor看成一个高维数组的话,slice就是返回它的一个子集
参数
tf.slice(
input_,
begin,
size,
name=None
)
用例
t = tf.constant([[[1, 1, 1], [2, 2, 2]],
[[3, 3, 3], [4, 4, 4]],
[[5, 5, 5], [6, 6, 6]]])
tf.slice(t, [1, 0, 0], [1, 1, 3]) # [[[3, 3, 3]]]
tf.slice(t, [1, 0, 0], [1, 2, 3]) # [[[3, 3, 3],
# [4, 4, 4]]]
tf.slice(t, [1, 0, 0], [2, 1, 3]) # [[[3, 3, 3]],
# [[5, 5, 5]]]
- begin,size是来个长度相同的列表。它们的长度是由input的维度决定的,begin[i] 对应着Tensor的第i维
- begin[I]代表着Tensor input 的第I维从begin[I]开始截取,截取长度由size[I]决定
- 返回是一个Tensor,不是Variable