scipy.linalg.blas.

get_blas_funcs

scipy.linalg.blas.get_blas_funcs(names, arrays=(), dtype=None, ilp64=False)

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

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

参数:

名称str 或 str 序列

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

数组ndarrays 序列，可选

可以提供数组来确定 BLAS 例程的最佳前缀。如果没有提供，将使用双精度例程，否则将使用数组中最通用的类型。

dtypestr 或 dtype, 可选

数据类型说明符。如果 arrays 非空，则不使用。

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

是否返回 ILP64 例程变体。选择 ‘preferred’ 将返回可用的 ILP64 例程，否则返回 32 位例程。默认值：False

返回:

函数列表

包含找到的函数列表。

注释

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

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

示例

Try it in your browser!

>>> 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'

Go BackOpen In Tab