vector_inner_product

Renvoie le produit scalaire entre deux vecteurs flottants. Les vecteurs doivent avoir la même dimension.

Pour la fonction Databricks SQL correspondante, consultez la fonctionvector_inner_product.

Syntaxe

Python

from pyspark.sql import functions as dbf

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

parameter

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.

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.

Renvoie

pyspark.sql.Column: produit interne en tant que valeur flottante.

Exemples

Python

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