获取属性错误 TensorDataset object 没有属性“输出形状”问题

Question

Getting attribute error issue using tf.data.Dataset

import numpy as np
import tensorflow as tf

x, y = np.array([1, 2, 3, 4]), np.array([5, 6, 7, 8])
d = tf.data.Dataset.from_tensors((x,y))

print(d.output_shapes) 

AttributeError: 'TensorDataset' object has no attribute 'output_shapes'

Screenshot

How to find output shapes?

Answer 1

Use tf.data.Dataset.elementspec :

import numpy as np
import tensorflow as tf

x, y = np.array([1, 2, 3, 4]), np.array([5, 6, 7, 8])
d = tf.data.Dataset.from_tensors((x,y))
d.element_spec

(TensorSpec(shape=(4,), dtype=tf.int64, name=None),
 TensorSpec(shape=(4,), dtype=tf.int64, name=None))