Tensorflow 如何将MapDataset转换为张量

我使用tensorflow\u io\u bigquery\u客户端从bq读取数据。 每个记录有4个整数元素

但是我需要把它转换成张量（每个4个整数）

所以我有点像：

tensorflow_io_bigquery_client = BigQueryClient()
read_session = tensorflow_io_bigquery_client.read_session(..)

dataset = read_session.parallel_read_rows()
dataset = dataset.map (transofrom_row)
#todo convert it into tensor (but i guess it should be an iterator or something like this because the dataset is hugh?

k_means_estimator = tf.compat.v1.estimator.experimental.KMeans(num_clusters = num_clusters, use_mini_batch = False, relative_tolerance = 1)

fit = k_means_estimator.train(input_fn=lambda: to_tensor(dataset), steps=1000) 
# todo convert dataset into tensor

---

我想问同样的问题，有什么办法吗？导师……确实可以帮助您解决问题。谢谢