tf.data.Dataset expands dim in model.fit if batch_input_shape is (None,)

General Discussion
, ,
#1

I created tf.data.Dataset object with shape ({‘users’:(None,), ‘movies’:(None,)}, (None,)). If I iterate over the dataset object outside model.fit, I see that the shape is same as specified above. But inside model.fit, the input shape changes to {‘users’:(None,1), ‘movies’:(None,1)]. I am using model sub-classing.

#3

Hi @tyagi619

Welcome to the TensorFlow Forum!

The given information is not sufficient to understand the issue. Could you please provide minimal reproducible code to replicate the error? Thank you.