如何获取tensorflow.data.Dataset的行数，列数/维数? [英] How to get number of rows, columns /dimensions of tensorflow.data.Dataset?

问题描述

像pandas_df.shape一样，tensorflow.data.Dataset有什么办法吗? 谢谢.

Like pandas_df.shape is there any way for tensorflow.data.Dataset? Thanks.

推荐答案

我不熟悉内置的东西，但是可以从Dataset._tensors属性中检索形状.示例:

I'm not familiar with something built-in, but the shapes could be retrieved from Dataset._tensors attribute. Example:

import tensorflow as tf

def dataset_shapes(dataset):
    try:
        return [x.get_shape().as_list() for x in dataset._tensors]
    except TypeError:
        return dataset._tensors.get_shape().as_list()

和用法:

from sklearn.datasets import make_blobs

x_train, y_train = make_blobs(n_samples=10,
                              n_features=2,
                              centers=[[1, 1], [-1, -1]],
                              cluster_std=0.5)
dataset = tf.data.Dataset.from_tensor_slices(x_train)
print(dataset_shapes(dataset)) # [10, 2]

dataset = tf.data.Dataset.from_tensor_slices((x_train, y_train))
print(dataset_shapes(dataset)) # [[10, 2], [10]]

这篇关于如何获取tensorflow.data.Dataset的行数，列数/维数?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！