Как получить количество строк, столбцов / размеров tenorflow.data.Dataset?

Question

Как и pandas_df.shape, есть ли способ для tenorflow.data.Dataset? Спасибо.

Answer 1

Я не знаком с чем-то встроенным, но формы могут быть получены из атрибута Dataset._tensors.Пример:

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]]