code: def _strip_first_and_last_dimension(x, batch_size):
      return tf.reshape(x[0, :], (batch_size,))
  
file_path: autodist/examples/benchmark/utils/recommendation/neumf_model.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import sys
  from six.moves import xrange
  import tensorflow
  from utils.recommendation import constants
  from utils.recommendation import movielens
  from utils.recommendation import ncf_common
  from utils.recommendation import stat_utils
  from utils.logs import mlperf_helper
label_desc: Use image node_id 3 for calling a global function with example usage: _strip_first_and_last_dimension(x, batch_size) and returns: tf