このページの翻訳は最新ではありません。ここをクリックして、英語の最新版を参照してください。
加算を伴うストライド バッチ行列乗算演算の最適化された GPU 実装
は、行列のバッチについてストライド行列-行列乗算と加算を実行します。バッチの各インスタンスの入力行列 D
= gpucoder.stridedMatrixMultiplyAdd(A
,B
,C
)A
、B
、および C
は、前のインスタンスのアドレスと同じアドレス オフセットに位置します。関数 gpucoder.stridedMatrixMultiplyAdd
は次の形式の行列-行列乗算を実行します。
ここで、 および はスカラー乗算係数、A
、B
、C
、および D
はそれぞれ m
行 k
列、k
行 n
列、m
行 n
列、および m
行 n
列の次元をもつ行列です。A
および B
は、転置またはエルミート共役にすることもできます。既定の設定では、 および は 1 に設定され、行列は転置されません。さまざまなスカラー乗算係数を指定し、入力行列の転置演算を実行するには、Name,Value
ペアの引数を使用します。
関数 gpucoder.stridedMatrixMultiplyAdd
に渡されるすべてのバッチは一様でなければなりません。つまり、すべてのインスタンスは同じ次元 m,n,k
でなければなりません。
___ = gpucoder.stridedMatrixMultiplyAdd(___,
は、1 つ以上の Name,Value
)Name,Value
ペアの引数で指定されたオプションを使用することにより、バッチ行列乗算演算と加算演算を実行します。
codegen
| coder.gpu.kernel
| coder.gpu.kernelfun
| gpucoder.batchedMatrixMultiply
| gpucoder.batchedMatrixMultiplyAdd
| gpucoder.stridedMatrixMultiply