Не зная точного контекста, на это довольно сложно ответить. Но в целом в модельных обучающих приложениях первое измерение является заполнителем для размера пакета, который обычно является переменным и известен только во время выполнения. Однако размер данных, с которыми вы работаете, обычно предопределен и должен быть закодирован в график для выполнения.
Так что, если state_dim
представляет и изображение с формой (300, 300, 3)
, то получающаяся форма заполнителя становится (None, 300, 300, 3)