How to get cosine similarity scores for all users and all the items in PySpark, if user's and item's embeddings are given?

Question

I have a users df-

df1 = spark.createDataFrame([
    ("u1", [0., 2., 3.]),
    ("u2", [1., 0., 0.]),
    ("u3", [0., 0., 3.]),