在Tensorflow中需要Caffe切片层

我定义了下面的切片层，其中我想将Nx7拆分为Nx3和Nx4两个blob

layers {
name: "slice_label"
type: SLICE
bottom: "label"
top: "label_pos"
top: "label_rotation"
slice_param {
    slice_dim: 1
    slice_point: 3
}

---

如何在Tensorflow中执行相同的操作

我相信您可以使用手工制作的图层，使用。您还可以通过单独调用slice来完成此操作。差不多

slice3, slice4 = tf.split(input_tensor, [3, 4], 1)

---

每个OP评论的更正：

slice3, slice4 = tf.split_v(labels, [3, 4], 1)

顺便说一句，间距遵循PEP8标准。当然，这不是必需的，但是有一个检查软件更喜欢这样的间距。

太好了！这一个适用于我的“slice3，slice4=tf.split_v（标签，[3,4]，1）”。请更新您的答案，我会将其标记为正确！谢谢