有什么办法可以打印进入Keras模型的输入? 我不确定“ batch_size”是指样本数还是每个样本的数据点数。如果可以将输入的数据打印到Keras模型中,那就太好了。
答案 0 :(得分:0)
批次大小是一步将通过网络的样本数。因此,将步数与批次大小相乘即可得出样本总数。
批量大小=样本数/步数
获取Keras模型的输入张量:
input_tensor = model.input
要从input_tensor
获取形状张量:
shape_tensor = tf.shape( input_tensor )
通过向shape_tensor
馈送一些输入,在tf.Session()
中运行input_tensor
:
with tf.Session() as session:
print( session( shape_tensor , feed_dict={ input_tensor:some_input } ) )
这将打印形状。