code: def get_output_shape(
      self, Q_input_shape, state_shape_1, state_shape_2
  ):
      n_samples = Q_input_shape[0]
      n_states = state_shape_1[1]
      n_filters = Q_input_shape[-1]
      return tf.TensorShape((n_samples * n_states, n_filters))
  
imports: import os
  import argparse
  from functools import partial
  import tensorflow
  from neupy import init, algorithms, storage
  from neupy.layers import *
  from neupy.utils import as_tuple, asfloat, tf_utils, tensorflow_session
  from loaddata import load_data
  from settings import MODELS_DIR, environments
  from evaluation import evaluate_accuracy
label_desc: Use image node_id 1 for calling the SelectValueAtStatePosition obj's underlying member method code with example usage: obj.get_output_shape(Q_input_shape, state_shape_1, state_shape_2) and returns: tf