在数据集上使用`batch()`后,张量形状对我来说没有意义

时间:2018-08-17 16:35:43

标签: tensorflow dataset

这是我测试tf.data.Dataset的代码:

[1,10,10]

程序输出的形状为[3,10,10]而不是 dependencies: flutter: sdk: flutter cupertino_icons: ^0.1.2 rxdart: ^0.18.0 ,这对我完全没有意义。

任何人都可以帮助解释这一点吗?

谢谢!

1 个答案:

答案 0 :(得分:1)

根据the docs{ "stress": { "fxPOS": { "sx": [ -14.2581, -15.6962, -12.6547, -13.639, -6.3305, 7.0567, 15.508 ] } } } 使用给定输入中的单个元素创建一个数据集。也就是说,您正在创建一个数据元素为from_tensors形状的数据集。
由于您只有一个元素,因此[10, 10]只能创建一个元素的批处理,从而提供一个新的数据集,其中包含一个形状为batch的元素。

也许您正在寻找[1, 10, 10]?在这种情况下,这将创建一个包含10个元素的数据集,每个元素的形状为tf.data.Dataset.from_tensor_slices。然后,应用[10]实际上会给您3个批次,即形状为batch(3)的元素(剩余的批次中只有一个元素)。