CUDA cuBLAS Level1函数介绍

类定义:class numbapro.cudalib.cublas.Blas(*args, **kws)

使用:这里写图片描述

level 1 (参数均为1维向量)

amax(x)

功能:返回一维数组中最大值的索引
使用方法:
CUDA cuBLAS Level1函数介绍_第1张图片

注意事项:数组中的数据不可以是整型,不然会报错。

amin(x):

同amax(x)

asum(x):

同amax(x)

axpy(alpha, x, y)

功能:y = alpha*x + y
使用方法:

nrm2(x)

功能:求向量x的2-范数
p-范数:若 ,那么

dot(x, y)

功能:求x和y的内积

dotc(x, y)

功能:当x为复数时,计算x的共轭与y的内积
使用:
CUDA cuBLAS Level1函数介绍_第2张图片

scal(alpha, x)

功能:x = alpha*x

rot(x, y, c, s)

功能:旋转,(x, y) = (c*x + s*y, -s*x + c*y)
使用:

rotg(a, b)

功能:返回一个数组tuple = (r, z, c, s),其中,

      r = sqrt(a**2 + b**2)
      z : 用于恢复c和s
           if abs(z) < 1:
                c, s = 1 - z ** 2, z
           elif abs(z) == 1:
                c, s = 0, 1
           else:
                c, s = 1 / z, 1 - z ** 2
      c = cos (向量[a, b]的cos值)
      s = sin  (向量[a, b]的cos值)

使用:

rotm(x, y, param)

功能:param — [flag, h11, h21, h12, h22]
x, y = h11 * x + h12 * y, h21 * x + h22 * y
旋转矩阵:

关于flag:
这里写图片描述
使用:
CUDA cuBLAS Level1函数介绍_第3张图片

rotmg(d1, d2, x1, y1)

功能:返回矩阵H,H*这里写图片描述 = (r, 0)
使用:
这里写图片描述

你可能感兴趣的:(CUDA)