我希望在
tf.data
中切片数据集。我的数据如下所示:dataset = tf.data.Dataset.from_tensor_slices([[0, 1, 2, 3, 4],
[1, 2, 3, 4, 5],
[2, 3, 4, 5, 6],
[3, 4, 5, 6, 7],
[4, 5, 6, 7, 8]])
然后主要的数据是:
[0 1 2 3 4]
[1 2 3 4 5]
[2 3 4 5 6]
[3 4 5 6 7]
[4 5 6 7 8]
我想创建另一个张量数据集,其中包含像这样的数据:
[[1, 2],
[2, 3],
[3, 4],
[4, 5],
[5, 6]]
在numpy中,它是这样的:
dataset[:,1:3]
如何在TensorFlow中做到这一点?
更新:
我使用以下方法实现了这个:
dataset2 = dataset.map(lambda data: data[1:3])
for val in dataset2:
print(val.numpy())
但我认为有好的解决方案。