vector_inner_product

Retourne le produit interne (produit par points) entre deux vecteurs float. Les vecteurs doivent avoir la même dimension.

Pour obtenir plus de détails sur la fonction SQL de Databricks correspondante, consultez vector_inner_product.

Syntax

from pyspark.sql import functions as dbf

dbf.vector_inner_product(left=<left>, right=<right>)

Parameters

Parameter Type Description
left pyspark.sql.Column ou nom de colonne Première colonne vectorielle.
right pyspark.sql.Column ou nom de colonne Deuxième colonne vectorielle.

Returns

pyspark.sql.Column: produit interne sous forme de valeur flottante.

Exemples

from pyspark.sql import functions as dbf
from pyspark.sql.types import ArrayType, FloatType, StructType, StructField

schema = StructType([StructField('a', ArrayType(FloatType())), StructField('b', ArrayType(FloatType()))])
df = spark.createDataFrame([([1.0, 2.0, 3.0], [4.0, 5.0, 6.0])], schema)
df.select(dbf.vector_inner_product('a', 'b')).first()[0]
# 32.0