TensorFlow的Dataset的padded_batch使用
padded_batch API如下padded_batch( batch_size, padded_shapes=None, padding_values=None, drop_remainder=False)注意参数drop_remainder用来约束最后一个batch是不是要丢掉,当这个batch样本数少于batch_size时,比如batch_size = 3,最后一个batch只有2个样本。默认是不丢掉padded_batch是非常见的一个操作,比如对一个变成的...
原创
2020-08-16 00:25:37 ·
4197 阅读 ·
1 评论