code: def _logmatmulexp(a, b):
      """`matmul` computed in log space."""
      return tf.reduce_logsumexp(
          a[..., :, :, None] + b[..., None, :, :], axis=-2
      )
  
imports: import hypothesis
  from hypothesis.extra import numpy
  import hypothesis.strategies
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.marginalize.logeinsumexp import _binary_einslogsumexp
  from tensorflow_probability.python.experimental.marginalize.logeinsumexp import logeinsumexp
  from tensorflow_probability.python.internal import test_util
  import tensorflow_probability.python.internal.hypothesis_testlib
label_desc: Use image node_id 1 for calling a global function with example usage: _logmatmulexp(a, b) and returns: tf