Python pyspark DenseVector.dot用法及代码示例

本文简要介绍 pyspark.ml.linalg.DenseVector.dot 的用法。

用法: dot(other)

计算两个向量的点积。我们支持(Numpy 数组、列表、SparseVector 或 SciPy 稀疏)和一维或二维目标 NumPy 数组。相当于调用两个向量的numpy.dot。

例子：

>>> dense = DenseVector(array.array('d', [1., 2.]))
>>> dense.dot(dense)
5.0
>>> dense.dot(SparseVector(2, [0, 1], [2., 1.]))
4.0
>>> dense.dot(range(1, 3))
5.0
>>> dense.dot(np.array(range(1, 3)))
5.0
>>> dense.dot([1.,])
Traceback (most recent call last):
    ...
AssertionError: dimension mismatch
>>> dense.dot(np.reshape([1., 2., 3., 4.], (2, 2), order='F'))
array([  5.,  11.])
>>> dense.dot(np.reshape([1., 2., 3.], (3, 1), order='F'))
Traceback (most recent call last):
    ...
AssertionError: dimension mismatch

相关用法

注：本文由纯净天空筛选整理自spark.apache.org大神的英文原创作品 pyspark.ml.linalg.DenseVector.dot。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。