举例
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]]]
解释
首先t的shape是(3,2,3)。
slice的函数输入分别是:
t:被切割对象
[1,0,0]:每个维度切割起始位置
[1,1,3]:每个维度切割数目
因此输出是[[[3, 3, 3]]]
参考:https://www.jianshu.com/p/71e6ef6c121b