code: def gram_style(a):
      with tf.name_scope("gram"):
          chn = int(a.shape[-1])
          a = tf.reshape(a, [-1, chn])
          n = tf.shape(a)[0]
          gram = tf.matmul(a, a, transpose_a=True)
          return gram / tf.cast(n, tf.float32)
  
file_path: lucid/lucid/optvis/style.py
imports: import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: gram_style(a) and returns: unknown