我正在将一批图像传递给我的神经网络。假设一批的形状是(4, 224, 224, 3)
。现在我想对这批数据进行切片操作,以便分别得到两个形状为(2, 224, 224, 3)
的张量。我该如何使用tf.slice()
或类似的方法来实现这一点?
回答:
我认为你更应该使用tf.split
。例如,在你的情况下,
tf.split(my_tensor, 2)
我正在将一批图像传递给我的神经网络。假设一批的形状是(4, 224, 224, 3)
。现在我想对这批数据进行切片操作,以便分别得到两个形状为(2, 224, 224, 3)
的张量。我该如何使用tf.slice()
或类似的方法来实现这一点?
回答:
我认为你更应该使用tf.split
。例如,在你的情况下,
tf.split(my_tensor, 2)