Размер партии - это количество выборок, которые пройдут через сеть за один шаг.Следовательно, при умножении количества шагов на размер пакета выдается общее количество выборок.
batch_size = number_of_samples / number_of_steps
Чтобы получить входной тензор вашей модели 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 } ) )
Это напечатает форму.