scipy.linalg.

get_blas_funcs#

scipy.linalg.get_blas_funcs(names, arrays=(), dtype=None, ilp64=False)[source]#

从名称返回可用的 BLAS 函数对象。

数组用于确定 BLAS 例程的最优前缀。

参数:
namesstr 或 str 序列

没有类型前缀的 BLAS 函数名称。

arraysndarray 序列,可选

可以指定数组以确定 BLAS 例程的最优前缀。若未指定,则将使用双精度例程,否则将使用数组中最通用的类型。

dtypestr 或 dtype,可选

数据类型说明符。若arrays 非空,则不使用。

ilp64{True, False, ‘preferred’},可选

是否返回 ILP64 例程变量。选择“preferred”在可用时返回 ILP64 例程,否则返回 32 位例程。默认值: False

返回:
funcs列表

包含找到的函数的列表。

说明

此例程在 Fortran/C 接口之间自动选择。对于以列优先顺序排列的数组,尽可能使用 Fortran 代码。在所有其他情况下,优先使用 C 代码。

在 BLAS 中,命名约定是所有函数都以类型前缀开头,该前缀取决于主矩阵的类型。对于 NumPy 类型 {float32, float64, complex64, complex128},这些类型可以是 {‘s’, ‘d’, ‘c’, ‘z’} 之一。代码和 dtype 存储在返回函数的属性 typecodedtype 中。

示例

>>> import numpy as np
>>> import scipy.linalg as LA
>>> rng = np.random.default_rng()
>>> a = rng.random((3,2))
>>> x_gemv = LA.get_blas_funcs('gemv', (a,))
>>> x_gemv.typecode
'd'
>>> x_gemv = LA.get_blas_funcs('gemv',(a*1j,))
>>> x_gemv.typecode
'z'