在Tensorflow中需要切片Caffe层

Question

我定义了以下切片图层，我想将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中做同样的事情？

Answer 1

我相信您可以使用split method使用手工制作的图层执行此操作。您也可以通过单独调用切片来执行此操作。像

这样的东西

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

每个OP评论的更正：

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

BTW，间距遵循PEP8标准。当然，这不是必需的，但有一种检查软件更喜欢这种间距。