标签： machine-learning tensorflow neural-network deep-learning conv-neural-network
我正在将一批图像传递给我的神经网络。我们假设批次的形状为(4, 224,224,3)。现在我想对我的批处理应用切片操作，这样我就可以得到两个形状为(2,224,224,3)的张量。我如何使用tf.slice()或类似的东西来做到这一点？
(4, 224,224,3)
(2,224,224,3)
tf.slice()
答案 0 :(得分：0)
我认为您更愿意使用tf.split。例如，在你的情况下，
tf.split
tf.split(my_tensor, 2)