x = tf.data.Dataset.from_tensor_slices(tf.float32, [None, 64*64]) y = tf.data.Dataset.from_tensor_slices(tf.float32, [None, num_classes])正确吗
时间: 2024-01-16 21:02:33 浏览: 146
TensorFlow数据读取
不正确。应该使用以下代码:
```
x = tf.data.Dataset.from_tensor_slices(tf.zeros([None, 64, 64], dtype=tf.float32))
y = tf.data.Dataset.from_tensor_slices(tf.zeros([None, num_classes], dtype=tf.float32))
```
这里的 `tf.zeros` 函数会创建一个张量,所有元素都是0。第一个张量的形状是 `[None, 64, 64]`,表示输入数据的形状是 `(batch_size, 64, 64)`。第二个张量的形状是 `[None, num_classes]`,表示输出数据的形状是 `(batch_size, num_classes)`。这里的 `None` 表示 batch_size 的大小是可变的。
阅读全文