code: def direction(layer, vec, cossim_pow=0, batch=None):
      """Visualize a direction"""
      vec = vec[None, None, None]
      vec = vec.astype("float32")
  
      @handle_batch(batch)
      def inner(T):
          return _dot_cossim(T(layer), vec, cossim_pow=cossim_pow)
  
      return inner
  
imports: from __future__ import absolute_import, division, print_function
  from decorator import decorator
  import numpy
  import tensorflow
  from lucid.optvis.objectives_util import _dot, _dot_cossim, _extract_act_pos, _make_arg_str, _T_force_NHWC, _T_handle_batch
label_desc: Use image node_id 5 for calling a global function with example usage: direction(layer, vec, cossim_pow, batch) and returns: inner, _dot_cossim