|
|
◆ csr_dusmm()
| def csr_dusmm |
( |
trans |
, |
|
|
order |
, |
|
|
m |
, |
|
|
n |
, |
|
|
l |
, |
|
|
alpha |
, |
|
|
val |
, |
|
|
rowptr |
, |
|
|
colind |
, |
|
|
base |
, |
|
|
b |
, |
|
|
beta |
, |
|
|
c |
|
|
) |
| |
C <- αAB + βC または C <- αATB + βC (CSR)
- 目的
- CSC形式の疎行列 A について次の行列×ベクトル演算を行う.
C <- αAB + βC または C <- αA^TB + βC
ここで, αおよびβはスカラー, A または A^T は m×n 疎行列, B は n×l 密行列, また, C は m×l 密行列である.
- 戻り値
- info (int)
= 0: 正常終了.
= i < 0: (-i)番目の入力パラメータの誤り.
- 引数
-
| [in] | trans | 操作を指定.
= 'N': C <- αAB + βC.
= 'T' または 'C': C <- αA^TB + βC. |
| [in] | order | b および c の格納順序.
= 'C': 列優先.
= 'R': 行優先. |
| [in] | m | 行列 A または A^T の行数. (m >= 0) (m = 0 の場合, 処理を行わずに戻る) |
| [in] | n | 行列 A または A^T の列数. (n >= 0) (n = 0 の場合, 処理を行わずに戻る) |
| [in] | l | 行列 B および C の列数. (l >= 0) (l = 0 の場合, 処理を行わずに戻る) |
| [in] | alpha | スカラー α. |
| [in] | val | Numpy ndarray (1次元配列, float, nnz)
行列 A の非ゼロ要素の値. (nnz は非ゼロ要素数) |
| [in] | rowptr | Numpy ndarray (1次元配列, int32, m + 1)
行列 A の行ポインタ. |
| [in] | colind | Numpy ndarray (1次元配列, int32, nnz)
行列 A の列インデクス. (nnz は非ゼロ要素数) |
| [in] | base | rowptr および colind のインデクス形式.
= 0: 0-ベース(C形式): 開始インデクス値が 0.
= 1: 1-ベース(Fortran形式): 開始インデクス値が 1. |
| [in] | b | Numpy ndarray (2次元配列, float, n x l)
行列 B. |
| [in] | beta | スカラー β. |
| [in,out] | c | Numpy ndarray (2次元配列, float, m x l)
[in] 入力行列 C (beta の値が 0 の場合, c の入力値は設定不要).
[out] 出力行列 (= αAB + βC). |
|