similarity_timing.py

import time

# Generate random data
x = np.random.rand(1000, 25)
y = np.random.rand(50, 25)

# Time NumPy
start = time.time()
for _ in range(100):
  np_cosine_similarity(x, y)

print("np time =", time.time() - start)

# Time TensorFlow
start = time.time()
for _ in range(100):
  tf_cosine_similarity(x, y).numpy()

print("tf time =", time.time() - start)

print("similarity output equal:", np.allclose(np_cosine_similarity(x, y), tf_cosine_similarity(x, y).numpy()))