首页 > 解决方案 > 填充形状在 Keras 中是如何工作的?

问题描述

padded_shapes = ([None], ())
train_batches = train_data.shuffle(1000).padded_batch(10, padded_shapes=padded_shapes)
test_batches = test_data.shuffle(1000).padded_batch(10, padded_shapes=padded_shapes)
train_batch, train_labels = next(iter(train_batches))
print(train_batch.numpy())

这是来自这个例子: https ://www.tensorflow.org/tutorials/text/word_embeddings

关于这一点:

padded_batch(10, padded_shapes=padded_shapes)

实际填充的序列长度是多少?我对代码进行了几次调试,每次 train_batch 的形状都不同,尽管 batch_size 始终为 10。

padded_shapes = ([None], ()) 初始化为空。

标签: kerastf.keras

解决方案


推荐阅读