vecdot

计算两个数组的（向量）点积。

参数

名称	类型	描述	默认值
x1	类数组	输入稀疏数组	必需
x2	类数组	输入稀疏数组	必需
axis	int	要约简的轴。	-1

返回值

名称	类型	描述
out	联合[SparseArray, ndarray]	包含点积的稀疏数组或0维数组。

源代码位于 sparse/numba_backend/_common.py

def vecdot(x1, x2, /, *, axis=-1):
    """
    Computes the (vector) dot product of two arrays.

    Parameters
    ----------
    x1, x2 : array_like
        Input sparse arrays
    axis : int
        The axis to reduce over.

    Returns
    -------
    out : Union[SparseArray, numpy.ndarray]
        Sparse or 0-D array containing dot product.
    """
    ndmin = builtins.min((x1.ndim, x2.ndim))
    if not (-ndmin <= axis < ndmin) or x1.shape[axis] != x2.shape[axis]:
        raise ValueError("Shapes must match along `axis`.")

    if np.issubdtype(x1.dtype, np.complexfloating):
        x1 = np.conjugate(x1)

    return np.sum(x1 * x2, axis=axis, dtype=np.result_type(x1, x2))