Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
tf.contrib.data.Dataset does not handle well with last elements with is fewer than batch size #13745
When this is that case
What would you recommend instead?
If you're using
dataset = dataset.padded_batch(batch_size, ...).filter(lambda x, y, z: tf.equal(tf.shape(x), batch_size))