AttributeError: ‘ShuffleDataset‘ object has no attribute ‘output_shapes‘ （tensorflow2.0 dataset ）

代码

train_dataset = train_dataset.padded_batch(batch_size,train_dataset.output_shapes)

 

异常提示

AttributeError                            Traceback (most recent call last)
<ipython-input-18-f554721ea273> in <module>
      3 
      4 train_dataset = train_dataset.shuffle(buffer_size)
----> 5 train_dataset = train_dataset.padded_batch(batch_size,train_dataset.output_shapes)
      6 #test_dataset = test_dataset.padded_batch(batch_szie,test_dataset.output_shapes)
      7 

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

 

猜测tensorflow2.0 api接口变更后没有这个属性字段

更改语句

train_dataset = train_dataset.padded_batch(batch_size,tf.compat.v1.data.get_output_shapes(train_dataset))