【发布时间】:2020-05-06 09:14:23
【问题描述】:
我正在使用 tf.keras API,我希望我的模型接受形状为 (None,) 的输入,None 是 batch_size。
keras.layers.Input() 的shape 不包含batch_size,所以我认为不能使用。
有没有办法实现我的目标?我更喜欢没有tf.placeholder 的解决方案,因为它已被弃用
顺便说一句,我的模型是一个句子嵌入模型,所以我希望输入类似于 ['How are you.','Good morning.']
======================= 更新:
目前,我可以使用layers.Input(dtype=tf.string,shape=1) 创建一个输入层,但这需要我的输入类似于[['How are you.'],['Good morning.']]。我希望我的输入只有一维。
【问题讨论】:
-
Keras 模型的输入总是有一个批处理维度,所以想要删除这个维度是没有意义的,框架不支持。
-
@MatiasValdenegro 我不想删除批量维度,我想要的是让我的输入成为一个简单的字符串。
-
一批字符串是有意义的,单个字符串不是一个好的输入(因为它错过了批量维度)。
-
是的,我的意思是训练时的一批字符串。
标签: tensorflow keras tensorflow2.0 tf.keras