more work

  y: tf.Tensor

    Input Tensor

  """

  x_reshape = tf.reshape(backend.sum(tf.square(x), axis=1), (-1, 1))

  y_reshape = tf.reshape(backend.sum(tf.square(y), axis=1), (1, -1))

  backend.dot(, ) + backend.epsilon()

  denom = backend.sqrt()

  x_norm = tf.nn.l2_normalize(x, axis=1)

  y_norm = tf.nn.l2_normalize(y, axis=1)

  return tf.math.divide(backend.dot(x, tf.transpose(y)), denom)

  return tf.reduce_sum(tf.multiply(x_norm,y_norm))

class AttnLSTMEmbedding(tf.keras.layers.Layer):

class TestLayers(test_util.TensorFlowTestCase):

  def test_cosine_dist(self):

    """Test invoking _cosine_dist."""

    x = np.ones((5, 4)).astype(np.float32)

    y = np.ones((5, 4)).astype(np.float32)

    assert layers._cosine_dist(x,y)

  def test_highway(self):

    """Test invoking Highway."""

    width = 5