code: def GaussianDistanceMatrix(R, rs, e):
      """Calculates gaussian distance matrix.
  
      B = batch_size, N = max_num_atoms, M = max_num_neighbors
  
      Parameters
      ----------
        R [B, N, M]: tf.Tensor
          Distance matrix.
        rs: tf.Variable
          Gaussian distance matrix mean.
        e: tf.Variable
          Gaussian distance matrix width (e = .5/std**2).
  
      Returns
      -------
        retval [B, N, M]: tf.Tensor
          Gaussian distance matrix.
  
      """
  
      return tf.exp(-e * (R - rs) ** 2)
  
file_path: deepchem/contrib/atomicconv/models/atomicnet_ops.py
imports: from __future__ import print_function
  from __future__ import division
  from __future__ import unicode_literals
  import numpy
  import tensorflow
label_desc: Use image node_id 6 for calling a global function with example usage: GaussianDistanceMatrix(R, rs, e) and returns: tf