import numpy as np
import tensorflow as tf
# Insert your own code for building `dataset`. For example:
dataset = tf.data.TFRecordDataset(...) # A dataset of tf.string records.
dataset = dataset.map(...) # Extract components from each tf.string record.
# Choose a value of `max_elems` that is at least as large as the dataset.
max_elems = np.iinfo(np.int64).max
dataset = dataset.batch(max_elems)
# Extracts the single element of a dataset as one or more `tf.Tensor` objects.
# No iterator needed in this case!
whole_dataset_tensors = tf.contrib.data.get_single_element(dataset)
# Create a session and evaluate `whole_dataset_tensors` to get arrays.
with tf.Session() as sess:
whole_dataset_arrays = sess.run(whole_dataset_tensors)
您可以使用^{} 转换和^{} 来完成此操作。
作为复习,
dataset.batch(n)
将占用n
个连续的n
个元素,并通过连接每个组件将它们转换为一个元素。这要求所有元素的每个组件都有一个固定的形状。如果n
大于dataset
中的元素数量(或者如果n
没有精确划分元素的数量),那么最后一批可以更小。因此,您可以为n
选择一个较大的值并执行以下操作:相关问题 更多 >
编程相关推荐