Commit 240ce321 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

yapf

parent 1fb1e5e2

deepchem/nn/model_ops.py

+2 −2
Original line number Diff line number Diff line
@@ -586,8 +586,8 @@ def cosine_distances(test, support): 
  tf.Tensor:

    Of shape (n_test, n_support)

  """

  rnorm_test = tf.rsqrt(

      tf.reduce_sum(tf.square(test), 1, keep_dims=True)) + 1e-7

  rnorm_test = tf.rsqrt(tf.reduce_sum(tf.square(test), 1,

                                      keep_dims=True)) + 1e-7

  rnorm_support = tf.rsqrt(

      tf.reduce_sum(tf.square(support), 1, keep_dims=True)) + 1e-7

  test_normalized = test * rnorm_test

CRA Git | Maintained and supported by SUSTech CRA and CCSE